Skip to content

Commit 8dea69d

Browse files
authored
Move getenv from kernel launch __call__ to __init__ (#3969)
Per comment #3907 (comment). `getenv` in each kernel launch would cost extra host time in the e2e model run, this change move it to `__init__` function to make it only gets called 1 time.
1 parent 6be08d4 commit 8dea69d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

third_party/intel/backend/driver.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -707,11 +707,11 @@ def __init__(self, src, metadata):
707707
self.signature = {idx: value for idx, value in src.signature.items()}
708708
src = make_launcher(self.constants, self.signature)
709709
self.mod = compile_module_from_src(src, "__triton_launcher")
710+
# Serialize KernelArguments for SPIR-V Runner
711+
self.serialize_kernel_args = os.getenv('TRITON_XPU_DUMP_SPIRV_KERNEL_ARGS', None)
710712

711713
def __call__(self, *args, **kwargs):
712-
# Serialize KernelArguments for SPIR-V Runner
713-
serialize_kernel_args = os.getenv('TRITON_XPU_DUMP_SPIRV_KERNEL_ARGS', None)
714-
if serialize_kernel_args:
714+
if self.serialize_kernel_args:
715715
serialize_args(args, self.constants, self.signature)
716716
self.mod.launch(args)
717717

0 commit comments

Comments
 (0)