Skip to content

Commit 7bcbafe

Browse files
committed
add target to make_ir for triton compatibility
1 parent 50eac81 commit 7bcbafe

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

torch/_higher_order_ops/triton_kernel_wrap.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -457,15 +457,15 @@ def get_signature_value(idx: int, arg: Any) -> str:
457457
inspect.signature(backend.get_codegen_implementation).parameters
458458
)
459459
if make_ir_sig_params == 2:
460-
ttir_module = src.make_ir(options, context)
460+
ttir_module = src.make_ir(target, options, context)
461461
elif make_ir_sig_params == 3:
462462
codegen_fns = backend.get_codegen_implementation()
463-
ttir_module = src.make_ir(options, codegen_fns, context)
463+
ttir_module = src.make_ir(target, options, codegen_fns, context)
464464
else:
465465
codegen_args = [options] if get_codegen_implementation_sig_params == 1 else []
466466
codegen_fns = backend.get_codegen_implementation(*codegen_args)
467467
module_map = backend.get_module_map()
468-
ttir_module = src.make_ir(options, codegen_fns, module_map, context)
468+
ttir_module = src.make_ir(target, options, codegen_fns, module_map, context)
469469
if not ttir_module.verify():
470470
raise RuntimeError("Verification for TTIR module has failed")
471471

0 commit comments

Comments
 (0)