Skip to content

Commit 3b29f6d

Browse files
Add scratch arguments to the kernel during launch (#5355)
Signed-off-by: Vaibhav Agarwal <[email protected]>
1 parent 0499edd commit 3b29f6d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

third_party/intel/backend/driver.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -662,6 +662,7 @@ def format_of(ty):
662662
// Submit the imported kernel.
663663
auto cgf = [&](sycl::handler &cgh) {{
664664
{" ".join(f'set_scalar_arg<{ty_to_cpp(item)}>(cgh, {idx}, params[{idx}]);' for idx, item in enumerate([signature[i] for i in signature if signature[i] != "constexpr"]))}
665+
{" ".join(f'set_scalar_arg<{ty_to_cpp(item)}>(cgh, {idx}, params[{idx}]);' for idx, item in enumerate(["*global_scratch", "*profile_scratch"], start=num_params-2))}
665666
if (shared_memory) {{
666667
using share_mem_t = sycl::local_accessor<int8_t, 1>;
667668
share_mem_t local_buffer = share_mem_t(shared_memory, cgh);

0 commit comments

Comments
 (0)