File tree Expand file tree Collapse file tree
third_party/mthreads/backend/spec/triton Expand file tree Collapse file tree Original file line number Diff line number Diff line change 3131
3232# flagtree backend specialization
3333from triton .flagtree_spec import spec
34+
3435spec ("triton_extend_globals" , globals ())
3536
3637__all__ = [
Original file line number Diff line number Diff line change 33from .compiler import compiler_extend_globals
44from .language import language_extend_globals
55
6+
67def triton_extend_globals (globals_dict ):
78 # NOTE: Must use absolute path import.
89 from triton .compiler .compiler import max_shared_mem
Original file line number Diff line number Diff line change 11def compiler_extend_globals (globals_dict ):
22 # NOTE: Must use absolute path import.
33 from triton .compiler .compiler import max_shared_mem
4- globals_dict ["max_shared_mem" ] = max_shared_mem
4+ globals_dict ["max_shared_mem" ] = max_shared_mem
Original file line number Diff line number Diff line change @@ -7,4 +7,4 @@ def language_extend_globals(globals_dict):
77 globals_dict ["unsqueeze" ] = unsqueeze
88 globals_dict ["_experimental_descriptor_load" ] = _experimental_descriptor_load
99 globals_dict ["_experimental_descriptor_store" ] = _experimental_descriptor_store
10- globals_dict ["to_tensor" ] = to_tensor
10+ globals_dict ["to_tensor" ] = to_tensor
You can’t perform that action at this time.
0 commit comments