Skip to content

Commit 8b0d167

Browse files
committed
fix different queue
1 parent a05d6ce commit 8b0d167

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sycl/chunked_prefill.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,8 @@ struct KernelRunner {
305305
cgh.parallel_for<KernelCur<FMHAChunkPrefillKernel>>(
306306
ConfigAccess.getRange(), ConfigAccess.getProperties(), KernelFunctor);
307307
};
308-
auto q = syclcompat::get_default_queue();
308+
auto stream = at::xpu::getCurrentXPUStream();
309+
auto q = stream.queue();
309310
q.submit(cgf).wait();
310311
// auto event = q.submit(cgf);
311312

0 commit comments

Comments
 (0)