From 6e09475283d9deb8aafb5a057935e5d058f927c2 Mon Sep 17 00:00:00 2001 From: chengjunlu Date: Wed, 27 Mar 2024 14:40:12 +0800 Subject: [PATCH] The shared information has been moved from binary to binary.metadata in Triton 3.0. Change the code to support that. (#3972) --- intel_extension_for_pytorch/_inductor/xpu/triton_heuristics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/intel_extension_for_pytorch/_inductor/xpu/triton_heuristics.py b/intel_extension_for_pytorch/_inductor/xpu/triton_heuristics.py index 960793820..7847fc0b7 100644 --- a/intel_extension_for_pytorch/_inductor/xpu/triton_heuristics.py +++ b/intel_extension_for_pytorch/_inductor/xpu/triton_heuristics.py @@ -212,7 +212,7 @@ def launcher({', '.join(def_args)}, grid, stream): launcher.config = cfg launcher.n_regs = getattr(binary, "n_regs", None) launcher.n_spills = getattr(binary, "n_spills", None) - launcher.shared = getattr(binary, "shared", None) + launcher.shared = scope["shared"] launcher.store_cubin = False # store this global varible to avoid the high overhead of reading it when calling run if launcher.store_cubin: