Integrate Triton up to [632bfc3](https://github.com/openai/triton/com…

…mits/632bfc342d3a7d63ce8b21209355139ee070d392) PiperOrigin-RevId: 715271136
jax-ml · Jan 15, 2025 · 8c18980 · 8c18980
1 parent e71d107
commit 8c18980
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax_triton/triton_lib.py b/jax_triton/triton_lib.py
@@ -395,7 +395,7 @@ def get_or_create_triton_kernel(
     context = _triton.ir.context()
     _triton.ir.load_dialects(context)
     backend.load_dialects(context)
-    codegen_fns = backend.get_codegen_implementation()
+    codegen_fns = backend.get_codegen_implementation(options)
 
     module = code_gen.ast_to_ttir(
         fn,