[Intel] Fix test_divide

Signed-off-by: Whitney Tsang <[email protected]>
intel · Dec 29, 2024 · 6919b06 · 6919b06
1 parent ab0745a
commit 6919b06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/test/regression/test_divide.py b/python/test/regression/test_divide.py
@@ -11,7 +11,7 @@
 def patch_kernel(template, to_replace):
     kernel = triton.JITFunction(template.fn)
     for key, value in to_replace.items():
-        kernel.src = kernel.src.replace(key, value)
+        kernel._unsafe_update_src(kernel.src.replace(key, value))
     return kernel