We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 28d04a5 commit 75302c5Copy full SHA for 75302c5
sycl/include/sycl/ext/oneapi/experimental/backend/cuda.hpp
@@ -69,6 +69,13 @@ interop_handle::get_native_context<backend::ext_oneapi_cuda>() const {
69
template <>
70
inline device make_device<backend::ext_oneapi_cuda>(
71
const backend_input_t<backend::ext_oneapi_cuda, device> &BackendObject) {
72
+ auto devs = device::get_devices(info::device_type::gpu);
73
+ for (auto &dev : devs) {
74
+ if (dev.get_backend() == backend::ext_oneapi_cuda &&
75
+ BackendObject == get_native<backend::ext_oneapi_cuda>(dev)) {
76
+ return dev;
77
+ }
78
79
pi_native_handle NativeHandle = static_cast<pi_native_handle>(BackendObject);
80
return ext::oneapi::cuda::make_device(NativeHandle);
81
}
0 commit comments