From 6919b06bfd55d3f94308dd98289e6388f1b443f7 Mon Sep 17 00:00:00 2001 From: Whitney Tsang Date: Sun, 29 Dec 2024 05:28:15 +0000 Subject: [PATCH] [Intel] Fix test_divide Signed-off-by: Whitney Tsang --- python/test/regression/test_divide.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/test/regression/test_divide.py b/python/test/regression/test_divide.py index 282b7b5c50..c8a4130ae1 100644 --- 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