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 a05d6ce commit 8b0d167Copy full SHA for 8b0d167
src/sycl/chunked_prefill.cpp
@@ -305,7 +305,8 @@ struct KernelRunner {
305
cgh.parallel_for<KernelCur<FMHAChunkPrefillKernel>>(
306
ConfigAccess.getRange(), ConfigAccess.getProperties(), KernelFunctor);
307
};
308
- auto q = syclcompat::get_default_queue();
+ auto stream = at::xpu::getCurrentXPUStream();
309
+ auto q = stream.queue();
310
q.submit(cgf).wait();
311
// auto event = q.submit(cgf);
312
0 commit comments