Skip to content

Commit 6919b06

Browse files
[Intel] Fix test_divide
Signed-off-by: Whitney Tsang <[email protected]>
1 parent ab0745a commit 6919b06

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/test/regression/test_divide.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
def patch_kernel(template, to_replace):
1212
kernel = triton.JITFunction(template.fn)
1313
for key, value in to_replace.items():
14-
kernel.src = kernel.src.replace(key, value)
14+
kernel._unsafe_update_src(kernel.src.replace(key, value))
1515
return kernel
1616

1717

0 commit comments

Comments
 (0)