-
Notifications
You must be signed in to change notification settings - Fork 63
Open
Open
Copy link
Milestone
Description
🚀 The feature, motivation and pitch
Existing:
torch-xpu-ops/src/comm/DeviceProperties.h
Lines 14 to 28 in b1582e1
| auto q = c10::xpu::getCurrentXPUStream(dev_id).queue(); | |
| auto ctx = q.get_context(); | |
| auto dev = q.get_device(); | |
| auto kid = ::sycl::get_kernel_id<KernelClass>(); | |
| // The kernel won't be built for devices except for the first device. | |
| // Launching kernel on devices except for the first device will raise | |
| // runtime error. Here is an alternative as a temporary solution to | |
| // provide an extra hint to SYCL runtime. | |
| // https://github.com/intel/llvm/issues/15127 | |
| auto kbundle = ::sycl::get_kernel_bundle<::sycl::bundle_state::executable>( | |
| ctx, {dev}, {kid}); | |
| ::sycl::kernel k = kbundle.get_kernel(kid); | |
| return k.get_info<::sycl::info::kernel_device_specific::work_group_size>(dev); |
TODO: intel/llvm#15650
Alternatives
No response
Additional context
No response