Add files using upload-large-folder tool

.gitattributes

@@ -34,3 +34,9 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 rustup-init.exe filter=lfs diff=lfs merge=lfs -text
+.rustup/downloads/371a98be3e239f760a7f0e87793ad0a871cdc2e607590cc47514d91fd8d1383c filter=lfs diff=lfs merge=lfs -text
+.rustup/downloads/8c0a40a5411746ff6600d93acd16652fef56702196bb25b940ff399d7e107f40 filter=lfs diff=lfs merge=lfs -text
+.venv/bin/z3.exe filter=lfs diff=lfs merge=lfs -text
+.venv/Lib/site-packages/30fcd23745efe32ce681__mypyc.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
+.venv/Lib/site-packages/PyWin32.chm filter=lfs diff=lfs merge=lfs -text
+.venv/Lib/site-packages/_cffi_backend.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text

.mypy_cache/3.13/_typeshed/__init__.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757957925,"dep_lines":[6,5,7,8,9,10,25,375,1,1,1,1],"dep_prios":[5,10,5,5,5,5,5,5,5,30,30,30],"dependencies":["collections.abc","sys","dataclasses","os","types","typing","typing_extensions","enum","builtins","_collections_abc","_frozen_importlib","abc"],"hash":"b0bcfcea11733df47eb95246cbea3442c7625c53","id":"_typeshed","ignore_all":true,"interface_hash":"293f4960090bfd50ecf661af53f1acf136ed9d81","mtime":1757955264,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Roaming\\Python\\Python313\\site-packages\\mypy\\typeshed\\stdlib\\_typeshed\\__init__.pyi","plugin_data":null,"size":13537,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/_typeshed/importlib.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"_typeshed.importlib","future_import_flags":[],"is_partial_stub_package":false,"is_stub":true,"names":{".class":"SymbolTable","LoaderProtocol":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"_typeshed.importlib.LoaderProtocol","name":"LoaderProtocol","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":["is_protocol"],"fullname":"_typeshed.importlib.LoaderProtocol","has_param_spec_type":false,"metaclass_type":"abc.ABCMeta","metadata":{},"module_name":"_typeshed.importlib","mro":["_typeshed.importlib.LoaderProtocol","builtins.object"],"names":{".class":"SymbolTable","load_module":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":[null,null],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"_typeshed.importlib.LoaderProtocol.load_module","name":"load_module","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":[null,null],"arg_types":["_typeshed.importlib.LoaderProtocol","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"load_module of LoaderProtocol","ret_type":"types.ModuleType","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"self_type":null,"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"MetaPathFinderProtocol":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"_typeshed.importlib.MetaPathFinderProtocol","name":"MetaPathFinderProtocol","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":["is_protocol"],"fullname":"_typeshed.importlib.MetaPathFinderProtocol","has_param_spec_type":false,"metaclass_type":"abc.ABCMeta","metadata":{},"module_name":"_typeshed.importlib","mro":["_typeshed.importlib.MetaPathFinderProtocol","builtins.object"],"names":{".class":"SymbolTable","find_spec":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,1],"arg_names":[null,null,null,null],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"_typeshed.importlib.MetaPathFinderProtocol.find_spec","name":"find_spec","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,1],"arg_names":[null,null,null,null],"arg_types":["_typeshed.importlib.MetaPathFinderProtocol","builtins.str",{".class":"UnionType","items":[{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"typing.Sequence"},{".class":"NoneType"}],"uses_pep604_syntax":true},{".class":"UnionType","items":["types.ModuleType",{".class":"NoneType"}],"uses_pep604_syntax":true}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"find_spec of MetaPathFinderProtocol","ret_type":{".class":"UnionType","items":["_frozen_importlib.ModuleSpec",{".class":"NoneType"}],"uses_pep604_syntax":true},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"self_type":null,"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"ModuleSpec":{".class":"SymbolTableNode","cross_ref":"_frozen_importlib.ModuleSpec","kind":"Gdef","module_hidden":true,"module_public":false},"ModuleType":{".class":"SymbolTableNode","cross_ref":"types.ModuleType","kind":"Gdef","module_hidden":true,"module_public":false},"PathEntryFinderProtocol":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"_typeshed.importlib.PathEntryFinderProtocol","name":"PathEntryFinderProtocol","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":["is_protocol"],"fullname":"_typeshed.importlib.PathEntryFinderProtocol","has_param_spec_type":false,"metaclass_type":"abc.ABCMeta","metadata":{},"module_name":"_typeshed.importlib","mro":["_typeshed.importlib.PathEntryFinderProtocol","builtins.object"],"names":{".class":"SymbolTable","find_spec":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,1],"arg_names":[null,null,null],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"_typeshed.importlib.PathEntryFinderProtocol.find_spec","name":"find_spec","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,1],"arg_names":[null,null,null],"arg_types":["_typeshed.importlib.PathEntryFinderProtocol","builtins.str",{".class":"UnionType","items":["types.ModuleType",{".class":"NoneType"}],"uses_pep604_syntax":true}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"find_spec of PathEntryFinderProtocol","ret_type":{".class":"UnionType","items":["_frozen_importlib.ModuleSpec",{".class":"NoneType"}],"uses_pep604_syntax":true},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"self_type":null,"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"Protocol":{".class":"SymbolTableNode","cross_ref":"typing.Protocol","kind":"Gdef","module_hidden":true,"module_public":false},"Sequence":{".class":"SymbolTableNode","cross_ref":"typing.Sequence","kind":"Gdef","module_hidden":true,"module_public":false},"__all__":{".class":"SymbolTableNode","kind":"Gdef","module_public":false,"node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"_typeshed.importlib.__all__","name":"__all__","setter_type":null,"type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"}}},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","module_public":false,"node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.importlib.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","module_public":false,"node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.importlib.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","module_public":false,"node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.importlib.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","module_public":false,"node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.importlib.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","module_public":false,"node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.importlib.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","module_public":false,"node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.importlib.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}}},"path":"C:\\Users\\marooc\\AppData\\Roaming\\Python\\Python313\\site-packages\\mypy\\typeshed\\stdlib\\_typeshed\\importlib.pyi"}

.mypy_cache/3.13/_typeshed/importlib.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757957925,"dep_lines":[4,5,6,7,1,1,1],"dep_prios":[5,5,5,5,5,30,30],"dependencies":["collections.abc","importlib.machinery","types","typing","builtins","_frozen_importlib","abc"],"hash":"fde3e178723c5fb47add8d7a7de403fe2ae224e8","id":"_typeshed.importlib","ignore_all":true,"interface_hash":"3a371054cf10cdeb3cc7f37aff649e6384794a1e","mtime":1757955264,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Roaming\\Python\\Python313\\site-packages\\mypy\\typeshed\\stdlib\\_typeshed\\importlib.pyi","plugin_data":null,"size":745,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/_typeshed/xml.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"_typeshed.xml","future_import_flags":[],"is_partial_stub_package":false,"is_stub":true,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef","module_hidden":true,"module_public":false},"DOMImplementation":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"_typeshed.xml.DOMImplementation","name":"DOMImplementation","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":["is_protocol"],"fullname":"_typeshed.xml.DOMImplementation","has_param_spec_type":false,"metaclass_type":"abc.ABCMeta","metadata":{},"module_name":"_typeshed.xml","mro":["_typeshed.xml.DOMImplementation","builtins.object"],"names":{".class":"SymbolTable","createDocument":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0],"arg_names":[null,null,null,null],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"_typeshed.xml.DOMImplementation.createDocument","name":"createDocument","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0],"arg_names":[null,null,null,null],"arg_types":["_typeshed.xml.DOMImplementation","builtins.str","builtins.str",{".class":"UnionType","items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"NoneType"}],"uses_pep604_syntax":true}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"createDocument of DOMImplementation","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"createDocumentType":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0],"arg_names":[null,null,null,null],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"_typeshed.xml.DOMImplementation.createDocumentType","name":"createDocumentType","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0],"arg_names":[null,null,null,null],"arg_types":["_typeshed.xml.DOMImplementation","builtins.str","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"createDocumentType of DOMImplementation","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"hasFeature":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":[null,null,null],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"_typeshed.xml.DOMImplementation.hasFeature","name":"hasFeature","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":[null,null,null],"arg_types":["_typeshed.xml.DOMImplementation","builtins.str",{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":true}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"hasFeature of DOMImplementation","ret_type":"builtins.bool","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"_typeshed.xml.DOMImplementation.Self","id":0,"name":"Self","namespace":"","upper_bound":"_typeshed.xml.DOMImplementation","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"Protocol":{".class":"SymbolTableNode","cross_ref":"typing.Protocol","kind":"Gdef","module_hidden":true,"module_public":false},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.xml.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.xml.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.xml.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.xml.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.xml.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"_typeshed.xml.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}}},"path":"C:\\Users\\marooc\\AppData\\Roaming\\Python\\Python313\\site-packages\\mypy\\typeshed\\stdlib\\_typeshed\\xml.pyi"}

.mypy_cache/3.13/_typeshed/xml.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757957938,"dep_lines":[3,1,1,1],"dep_prios":[5,5,30,30],"dependencies":["typing","builtins","_frozen_importlib","abc"],"hash":"736e174b6235a4093bfc69734db829f1c7651404","id":"_typeshed.xml","ignore_all":true,"interface_hash":"9f190c62c0974d498a44a58f1f9b9588875042da","mtime":1757955264,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Roaming\\Python\\Python313\\site-packages\\mypy\\typeshed\\stdlib\\_typeshed\\xml.pyi","plugin_data":null,"size":508,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/__init__.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__path__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.__path__","name":"__path__","setter_type":null,"type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"}}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\__init__.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/__init__.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757957994,"dep_lines":[1,1,1,1],"dep_prios":[5,30,30,30],"dependencies":["builtins","_frozen_importlib","abc","typing"],"hash":"da39a3ee5e6b4b0d3255bfef95601890afd80709","id":"torch._inductor.codegen.cuda","ignore_all":true,"interface_hash":"a0265f8049feb1e60fc6c509bd7cb81c4f70c9de","mtime":1757363944,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\__init__.py","plugin_data":null,"size":0,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cuda_kernel.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda.cuda_kernel","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"ArgInfo":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cuda_template.ArgInfo","kind":"Gdef"},"BaseSchedulerNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.scheduler.BaseSchedulerNode","kind":"Gdef"},"Buffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Buffer","kind":"Gdef"},"CSEVariable":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.CSEVariable","kind":"Gdef"},"CUDABenchmarkRequest":{".class":"SymbolTableNode","cross_ref":"torch._inductor.autotune_process.CUDABenchmarkRequest","kind":"Gdef"},"CUDAKernel":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":[{".class":"Instance","args":["torch._inductor.codegen.common.CSEVariable"],"extra_attrs":null,"type_ref":"torch._inductor.codegen.common.Kernel"}],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel","name":"CUDAKernel","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cuda_kernel","mro":["torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel","torch._inductor.codegen.common.Kernel","torch._inductor.codegen.common.CodeGen","builtins.object"],"names":{".class":"SymbolTable","__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,2,4],"arg_names":["self","args","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,2,4],"arg_names":["self","args","kwargs"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of CUDAKernel","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"add_layout_arg":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,0],"arg_names":["self","symbol","node","attr","dim"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.add_layout_arg","name":"add_layout_arg","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,0],"arg_names":["self","symbol","node","attr","dim"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutSymbols"},"torch._inductor.ir.IRNode",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutAttrs"},"builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"add_layout_arg of CUDAKernel","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"find_layout_arg":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0],"arg_names":["self","node","attr","dim"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.find_layout_arg","name":"find_layout_arg","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0],"arg_names":["self","node","attr","dim"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel","torch._inductor.ir.IRNode",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutAttrs"},"builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"find_layout_arg of CUDAKernel","ret_type":{".class":"UnionType","items":["torch._inductor.codegen.cuda.cuda_kernel.LayoutArg",{".class":"NoneType"}],"uses_pep604_syntax":false},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"find_ld_idx":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["node"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.find_ld_idx","name":"find_ld_idx","original_first_arg":"node","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["node"],"arg_types":["torch._inductor.ir.IRNode"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"find_ld_idx of CUDAKernel","ret_type":"builtins.int","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.find_ld_idx","name":"find_ld_idx","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0],"arg_names":["node"],"arg_types":["torch._inductor.ir.IRNode"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"find_ld_idx of CUDAKernel","ret_type":"builtins.int","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"find_symbol":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0],"arg_names":["self","node","attr","dim"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.find_symbol","name":"find_symbol","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0],"arg_names":["self","node","attr","dim"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel","torch._inductor.ir.IRNode",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutAttrs"},"builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"find_symbol of CUDAKernel","ret_type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_dynamic_shape_args":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.get_dynamic_shape_args","name":"get_dynamic_shape_args","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_dynamic_shape_args of CUDAKernel","ret_type":{".class":"Instance","args":[{".class":"UnionType","items":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cuda_kernel.Expr","source_any":null,"type_of_any":3},"builtins.int"],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_layout_args":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.get_layout_args","name":"get_layout_args","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_layout_args of CUDAKernel","ret_type":{".class":"Instance","args":[{".class":"UnionType","items":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cuda_kernel.Expr","source_any":null,"type_of_any":3},"builtins.int"],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"builtins.tuple"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"init_layout_args":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.init_layout_args","name":"init_layout_args","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"init_layout_args of CUDAKernel","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"layout_args":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.layout_args","name":"layout_args","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"Instance","args":["torch._inductor.codegen.cuda.cuda_kernel.LayoutArg"],"extra_attrs":null,"type_ref":"builtins.list"}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"named_nodes":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.named_nodes","name":"named_nodes","setter_type":null,"type":{".class":"Instance","args":["builtins.str","torch._inductor.ir.IRNode"],"extra_attrs":null,"type_ref":"builtins.dict"}}},"overrides":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.overrides","name":"overrides","setter_type":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.type","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":true,"is_ellipsis_args":false,"name":null,"ret_type":"torch._inductor.codegen.common.OpOverrides","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"size_args":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.size_args","name":"size_args","setter_type":null,"type":{".class":"Instance","args":[{".class":"UnionType","items":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cuda_kernel.Expr","source_any":null,"type_of_any":3},"builtins.int"],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"builtins.list"}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"CUDATemplate":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate","kind":"Gdef"},"CUDATemplateBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.CUDATemplateBuffer","kind":"Gdef"},"CUDATemplateCaller":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.ir.ChoiceCaller"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller","name":"CUDATemplateCaller","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cuda_kernel","mro":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller","torch._inductor.ir.ChoiceCaller","builtins.object"],"names":{".class":"SymbolTable","__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,0,0,0,0,0,0,0],"arg_names":["self","name","category","input_nodes","layout","make_kernel_render","bmreq","supports_epilogue_fusion","template","info_kwargs","description"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,0,0,0,0,0,0,0],"arg_names":["self","name","category","input_nodes","layout","make_kernel_render","bmreq","supports_epilogue_fusion","template","info_kwargs","description"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller","builtins.str","builtins.str",{".class":"Instance","args":["torch._inductor.ir.Buffer"],"extra_attrs":null,"type_ref":"builtins.list"},"torch._inductor.ir.Layout",{".class":"CallableType","arg_kinds":[0,0],"arg_names":[null,null],"arg_types":["torch._inductor.ir.CUDATemplateBuffer",{".class":"UnionType","items":[{".class":"Instance","args":["torch._inductor.scheduler.BaseSchedulerNode"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":{".class":"TupleType","implicit":false,"items":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel",{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"functools.partial"}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]},"torch._inductor.autotune_process.CUDABenchmarkRequest","builtins.bool","torch._inductor.codegen.cuda.cuda_template.CUDATemplate",{".class":"UnionType","items":[{".class":"Instance","args":["builtins.str",{".class":"UnionType","items":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"},{".class":"Instance","args":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"}],"extra_attrs":null,"type_ref":"builtins.list"}],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"builtins.dict"},{".class":"NoneType"}],"uses_pep604_syntax":false},"builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of CUDATemplateCaller","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"__str__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":[null],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.__str__","name":"__str__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":[null],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__str__ of CUDATemplateCaller","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"benchmark":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,2,3],"arg_names":["self","args","out"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.benchmark","name":"benchmark","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,2,3],"arg_names":["self","args","out"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"benchmark of CUDATemplateCaller","ret_type":"builtins.float","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"bmreq":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.bmreq","name":"bmreq","setter_type":null,"type":"torch._inductor.autotune_process.CUDABenchmarkRequest"}},"call_name":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.call_name","name":"call_name","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"call_name of CUDATemplateCaller","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"category":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.category","name":"category","setter_type":null,"type":"builtins.str"}},"hash_key":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.hash_key","name":"hash_key","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"hash_key of CUDATemplateCaller","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"info_dict":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.info_dict","name":"info_dict","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"info_dict of CUDATemplateCaller","ret_type":{".class":"Instance","args":["builtins.str",{".class":"UnionType","items":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"},{".class":"Instance","args":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"}],"extra_attrs":null,"type_ref":"builtins.list"}],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"builtins.dict"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"info_kwargs":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.info_kwargs","name":"info_kwargs","setter_type":null,"type":{".class":"UnionType","items":[{".class":"Instance","args":["builtins.str",{".class":"UnionType","items":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"},{".class":"Instance","args":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"}],"extra_attrs":null,"type_ref":"builtins.list"}],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"builtins.dict"},{".class":"NoneType"}],"uses_pep604_syntax":false}}},"kernel_hash_key":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.kernel_hash_key","name":"kernel_hash_key","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"kernel_hash_key of CUDATemplateCaller","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"make_kernel_render":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.make_kernel_render","name":"make_kernel_render","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":[null,null],"arg_types":["torch._inductor.ir.CUDATemplateBuffer",{".class":"UnionType","items":[{".class":"Instance","args":["torch._inductor.scheduler.BaseSchedulerNode"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":{".class":"TupleType","implicit":false,"items":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel",{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"functools.partial"}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"output_node":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.output_node","name":"output_node","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"output_node of CUDATemplateCaller","ret_type":"torch._inductor.ir.TensorBox","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"precompile":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.precompile","name":"precompile","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"precompile of CUDATemplateCaller","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"supports_epilogue_fusion":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.supports_epilogue_fusion","name":"supports_epilogue_fusion","setter_type":null,"type":"builtins.bool"}},"template":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.template","name":"template","setter_type":null,"type":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate"}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"CUDATemplateKernel":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","name":"CUDATemplateKernel","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cuda_kernel","mro":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.codegen.cuda.cuda_kernel.CUDAKernel","torch._inductor.codegen.common.Kernel","torch._inductor.codegen.common.CodeGen","builtins.object"],"names":{".class":"SymbolTable","_EXTRA_CPP_ARGS":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel._EXTRA_CPP_ARGS","name":"_EXTRA_CPP_ARGS","setter_type":null,"type":"builtins.str"}},"__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0],"arg_names":["self","kernel_name","runtime_arg_info","runtime_arg_values"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0],"arg_names":["self","kernel_name","runtime_arg_info","runtime_arg_values"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","builtins.str",{".class":"Instance","args":["torch._inductor.codegen.cuda.cuda_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of CUDATemplateKernel","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"batch_stride":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,1],"arg_names":["self","node","default_value"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.batch_stride","name":"batch_stride","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,1],"arg_names":["self","node","default_value"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.ir.IRNode","builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"batch_stride of CUDATemplateKernel","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"call_kernel":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","name","node"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.call_kernel","name":"call_kernel","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","name","node"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","builtins.str","torch._inductor.ir.CUDATemplateBuffer"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"call_kernel of CUDATemplateKernel","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"check_not_null":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","node"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.check_not_null","name":"check_not_null","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","node"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.ir.IRNode"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"check_not_null of CUDATemplateKernel","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"cutlass_dtype":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,1],"arg_names":["self","node","default_dtype"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.cutlass_dtype","name":"cutlass_dtype","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,1],"arg_names":["self","node","default_dtype"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.ir.IRNode",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"cutlass_dtype of CUDATemplateKernel","ret_type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"def_kernel":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,1,1],"arg_names":["self","inputs","outputs","names_str","input_reorder"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.def_kernel","name":"def_kernel","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,1,1],"arg_names":["self","inputs","outputs","names_str","input_reorder"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel",{".class":"Instance","args":["torch._inductor.ir.IRNode"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":["torch._inductor.ir.IRNode"],"extra_attrs":null,"type_ref":"builtins.list"},"builtins.str",{".class":"UnionType","items":[{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"def_kernel of CUDATemplateKernel","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"dtype":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","node"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.dtype","name":"dtype","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","node"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.ir.IRNode"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"dtype of CUDATemplateKernel","ret_type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_signature":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.get_signature","name":"get_signature","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_signature of CUDATemplateKernel","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"load":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,1],"arg_names":["self","name","index","mode"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.load","name":"load","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,1],"arg_names":["self","name","index","mode"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","builtins.str",{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cuda_kernel.Expr","source_any":null,"type_of_any":3},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"load of CUDATemplateKernel","ret_type":"torch._inductor.codegen.common.CSEVariable","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"max_valid_index":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,1],"arg_names":["self","node","default"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.max_valid_index","name":"max_valid_index","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,1],"arg_names":["self","node","default"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.ir.IRNode",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"max_valid_index of CUDATemplateKernel","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"offset":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","node"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.offset","name":"offset","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","node"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.ir.IRNode"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"offset of CUDATemplateKernel","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"ptr":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","node"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.ptr","name":"ptr","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","node"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.ir.IRNode"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"ptr of CUDATemplateKernel","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"row_or_column_stride":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,1],"arg_names":["self","node","default_value"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.row_or_column_stride","name":"row_or_column_stride","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,1],"arg_names":["self","node","default_value"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.ir.IRNode","builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"row_or_column_stride of CUDATemplateKernel","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"runtime_arg_info":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.runtime_arg_info","name":"runtime_arg_info","setter_type":null,"type":{".class":"Instance","args":["torch._inductor.codegen.cuda.cuda_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"}}},"runtime_arg_values":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.runtime_arg_values","name":"runtime_arg_values","setter_type":null,"type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"}}},"signature":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.signature","name":"signature","setter_type":null,"type":"builtins.str"}},"size":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,1,1],"arg_names":["self","node","start_index","end_index","default_value"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.size","name":"size","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,1,1],"arg_names":["self","node","start_index","end_index","default_value"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.ir.IRNode","builtins.int",{".class":"UnionType","items":["builtins.int",{".class":"NoneType"}],"uses_pep604_syntax":false},"builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"size of CUDATemplateKernel","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"store":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,1],"arg_names":["self","name","index","value","mode"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.store","name":"store","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,1],"arg_names":["self","name","index","value","mode"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","builtins.str",{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cuda_kernel.Expr","source_any":null,"type_of_any":3},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"store of CUDATemplateKernel","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"stride":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,1],"arg_names":["self","node","index","default_value"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.stride","name":"stride","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,1],"arg_names":["self","node","index","default_value"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","torch._inductor.ir.IRNode","builtins.int","builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"stride of CUDATemplateKernel","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"Callable":{".class":"SymbolTableNode","cross_ref":"typing.Callable","kind":"Gdef"},"ChoiceCaller":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.ChoiceCaller","kind":"Gdef"},"CppPrinter":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cpp_utils.CppPrinter","kind":"Gdef"},"CppWrapperCpu":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cpp_wrapper_cpu.CppWrapperCpu","kind":"Gdef"},"DTYPE_TO_CPP":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cpp_utils.DTYPE_TO_CPP","kind":"Gdef"},"DTYPE_TO_CUTLASS_TYPE":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cutlass_utils.DTYPE_TO_CUTLASS_TYPE","kind":"Gdef"},"Expr":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.Expr","name":"Expr","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cuda_kernel.Expr","source_any":null,"type_of_any":3}}},"IRNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.IRNode","kind":"Gdef"},"IndentedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.IndentedBuffer","kind":"Gdef"},"Kernel":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.Kernel","kind":"Gdef"},"Layout":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Layout","kind":"Gdef"},"LayoutArg":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg","name":"LayoutArg","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg","has_param_spec_type":false,"metaclass_type":null,"metadata":{"dataclass":{"attributes":[{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":65,"name":"node","type":"torch._inductor.ir.IRNode"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":66,"name":"symbol","type":{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutSymbols"}},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":67,"name":"attr","type":{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutAttrs"}},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":68,"name":"dim","type":"builtins.int"}],"frozen":true},"dataclass_tag":{}},"module_name":"torch._inductor.codegen.cuda.cuda_kernel","mro":["torch._inductor.codegen.cuda.cuda_kernel.LayoutArg","builtins.object"],"names":{".class":"SymbolTable","__dataclass_fields__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_classvar","is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.__dataclass_fields__","name":"__dataclass_fields__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"dataclasses.Field"}],"extra_attrs":null,"type_ref":"builtins.dict"}},"plugin_generated":true},"__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,0],"arg_names":["self","node","symbol","attr","dim"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,0],"arg_names":["self","node","symbol","attr","dim"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.LayoutArg","torch._inductor.ir.IRNode",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutSymbols"},{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutAttrs"},"builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of LayoutArg","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"__match_args__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_ready","allow_incompatible_override"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.__match_args__","name":"__match_args__","setter_type":null,"type":{".class":"TupleType","implicit":false,"items":[{".class":"LiteralType","fallback":"builtins.str","value":"node"},{".class":"LiteralType","fallback":"builtins.str","value":"symbol"},{".class":"LiteralType","fallback":"builtins.str","value":"attr"},{".class":"LiteralType","fallback":"builtins.str","value":"dim"}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}}},"plugin_generated":true},"__mypy-replace":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[5,5,5,5],"arg_names":["node","symbol","attr","dim"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.__mypy-replace","name":"__mypy-replace","original_first_arg":"node","type":{".class":"CallableType","arg_kinds":[5,5,5,5],"arg_names":["node","symbol","attr","dim"],"arg_types":["torch._inductor.ir.IRNode",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutSymbols"},{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutAttrs"},"builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-replace of LayoutArg","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_staticmethod","is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.__mypy-replace","name":"__mypy-replace","setter_type":null,"type":{".class":"CallableType","arg_kinds":[5,5,5,5],"arg_names":["node","symbol","attr","dim"],"arg_types":["torch._inductor.ir.IRNode",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutSymbols"},{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutAttrs"},"builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-replace of LayoutArg","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"plugin_generated":true},"__replace__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,5,5,5,5],"arg_names":["self","node","symbol","attr","dim"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.__replace__","name":"__replace__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,5,5,5,5],"arg_names":["self","node","symbol","attr","dim"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.LayoutArg","torch._inductor.ir.IRNode",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutSymbols"},{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutAttrs"},"builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__replace__ of LayoutArg","ret_type":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"attr":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_property","is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.attr","name":"attr","setter_type":null,"type":{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutAttrs"}}},"dim":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_property","is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.dim","name":"dim","setter_type":null,"type":"builtins.int"}},"matches":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0],"arg_names":["self","node","attr","dim"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.matches","name":"matches","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0],"arg_names":["self","node","attr","dim"],"arg_types":["torch._inductor.codegen.cuda.cuda_kernel.LayoutArg",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"matches of LayoutArg","ret_type":"builtins.bool","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"node":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_property","is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.node","name":"node","setter_type":null,"type":"torch._inductor.ir.IRNode"}},"symbol":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_property","is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.symbol","name":"symbol","setter_type":null,"type":{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutSymbols"}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cuda_kernel.LayoutArg","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"Literal":{".class":"SymbolTableNode","cross_ref":"typing.Literal","kind":"Gdef"},"OpOverrides":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.OpOverrides","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"OrderedSet":{".class":"SymbolTableNode","cross_ref":"torch.utils._ordered_set.OrderedSet","kind":"Gdef"},"Placeholder":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.Placeholder","kind":"Gdef"},"PrimitiveInfoType":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.PrimitiveInfoType","kind":"Gdef"},"TYPE_CHECKING":{".class":"SymbolTableNode","cross_ref":"typing.TYPE_CHECKING","kind":"Gdef"},"TensorBox":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.TensorBox","kind":"Gdef"},"Union":{".class":"SymbolTableNode","cross_ref":"typing.Union","kind":"Gdef"},"V":{".class":"SymbolTableNode","cross_ref":"torch._inductor.virtualized.V","kind":"Gdef"},"ValidLayoutAttrs":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeAlias","alias_tvars":[],"column":0,"fullname":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutAttrs","line":60,"no_args":false,"normalized":false,"python_3_12_type_alias":false,"target":{".class":"UnionType","items":[{".class":"LiteralType","fallback":"builtins.str","value":"size"},{".class":"LiteralType","fallback":"builtins.str","value":"stride"}],"uses_pep604_syntax":false}}},"ValidLayoutSymbols":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeAlias","alias_tvars":[],"column":0,"fullname":"torch._inductor.codegen.cuda.cuda_kernel.ValidLayoutSymbols","line":59,"no_args":false,"normalized":false,"python_3_12_type_alias":false,"target":{".class":"UnionType","items":[{".class":"LiteralType","fallback":"builtins.str","value":"M"},{".class":"LiteralType","fallback":"builtins.str","value":"N"},{".class":"LiteralType","fallback":"builtins.str","value":"K"},{".class":"LiteralType","fallback":"builtins.str","value":"B"},{".class":"LiteralType","fallback":"builtins.str","value":"lda"},{".class":"LiteralType","fallback":"builtins.str","value":"ldb"},{".class":"LiteralType","fallback":"builtins.str","value":"ldc"},{".class":"LiteralType","fallback":"builtins.str","value":"ldd"}],"uses_pep604_syntax":false}}},"ValueRanges":{".class":"SymbolTableNode","cross_ref":"torch.utils._sympy.value_ranges.ValueRanges","kind":"Gdef"},"WorkspaceArg":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.WorkspaceArg","kind":"Gdef"},"WorkspaceZeroMode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.WorkspaceZeroMode","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"_normalize_idx":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["index","total_length"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_kernel._normalize_idx","name":"_normalize_idx","original_first_arg":"index","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["index","total_length"],"arg_types":["builtins.int","builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_normalize_idx","ret_type":"builtins.int","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"cexpr":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.cexpr","name":"cexpr","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,5,5],"arg_names":["expr","simplify","p"],"arg_types":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"builtins.bool",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":true,"is_ellipsis_args":false,"name":null,"ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"config":{".class":"SymbolTableNode","cross_ref":"torch._inductor.config","kind":"Gdef"},"dataclass":{".class":"SymbolTableNode","cross_ref":"dataclasses.dataclass","kind":"Gdef"},"defaultdict":{".class":"SymbolTableNode","cross_ref":"collections.defaultdict","kind":"Gdef"},"do_bench_using_profiling":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.do_bench_using_profiling","kind":"Gdef"},"functools":{".class":"SymbolTableNode","cross_ref":"functools","kind":"Gdef"},"itertools":{".class":"SymbolTableNode","cross_ref":"itertools","kind":"Gdef"},"log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.log","name":"log","setter_type":null,"type":"logging.Logger"}},"logging":{".class":"SymbolTableNode","cross_ref":"logging","kind":"Gdef"},"symbols":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_kernel.symbols","name":"symbols","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cuda_kernel.symbols","source_any":null,"type_of_any":3}}},"sympy_product":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.sympy_product","kind":"Gdef"},"torch_dtype":{".class":"SymbolTableNode","cross_ref":"torch._C.dtype","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cuda_kernel.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cuda_kernel.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959417,"dep_lines":[18,22,13,16,36,44,11,14,15,24,25,35,11,2,3,4,5,6,7,11,1,1,1,1,1,1,1,1,1,1,1,1,9],"dep_prios":[5,20,5,5,5,5,10,5,5,5,5,5,20,10,10,10,5,5,5,5,5,30,30,30,30,30,30,30,30,30,30,30,5],"dependencies":["torch._inductor.codegen.cuda.cutlass_utils","torch._inductor.codegen.cuda.cuda_template","torch._inductor.codegen.cpp_wrapper_cpu","torch.utils._sympy.value_ranges","torch._inductor.codegen.common","torch._inductor.codegen.cpp_utils","torch._inductor.config","torch._inductor.scheduler","torch._inductor.utils","torch._inductor.autotune_process","torch._inductor.ir","torch._inductor.virtualized","torch._inductor","functools","itertools","logging","collections","dataclasses","typing","torch","builtins","_collections_abc","_frozen_importlib","_typeshed","abc","torch._C","torch._inductor.ops_handler","torch.utils","torch.utils._ordered_set","torch.utils._sympy","torch.utils._sympy.printers","types"],"hash":"26af886ddccaed42958dba8de2147f582c5a032e","id":"torch._inductor.codegen.cuda.cuda_kernel","ignore_all":true,"interface_hash":"b942c35836edde80e7123dbc4d39cf0d079cecf2","mtime":1757363945,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cuda_kernel.py","plugin_data":null,"size":24714,"suppressed":["sympy"],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cuda_template.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda.cuda_template","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"ArgInfo":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo","name":"ArgInfo","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo","has_param_spec_type":false,"metaclass_type":null,"metadata":{"dataclass":{"attributes":[{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":37,"name":"name","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":38,"name":"ty","type":"builtins.str"}],"frozen":true},"dataclass_tag":{}},"module_name":"torch._inductor.codegen.cuda.cuda_template","mro":["torch._inductor.codegen.cuda.cuda_template.ArgInfo","builtins.object"],"names":{".class":"SymbolTable","__dataclass_fields__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_classvar","is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo.__dataclass_fields__","name":"__dataclass_fields__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"dataclasses.Field"}],"extra_attrs":null,"type_ref":"builtins.dict"}},"plugin_generated":true},"__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","name","ty"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","name","ty"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.ArgInfo","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of ArgInfo","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"__match_args__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_ready","allow_incompatible_override"],"fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo.__match_args__","name":"__match_args__","setter_type":null,"type":{".class":"TupleType","implicit":false,"items":[{".class":"LiteralType","fallback":"builtins.str","value":"name"},{".class":"LiteralType","fallback":"builtins.str","value":"ty"}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}}},"plugin_generated":true},"__mypy-replace":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[5,5],"arg_names":["name","ty"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo.__mypy-replace","name":"__mypy-replace","original_first_arg":"name","type":{".class":"CallableType","arg_kinds":[5,5],"arg_names":["name","ty"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-replace of ArgInfo","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_staticmethod","is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo.__mypy-replace","name":"__mypy-replace","setter_type":null,"type":{".class":"CallableType","arg_kinds":[5,5],"arg_names":["name","ty"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-replace of ArgInfo","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"plugin_generated":true},"__replace__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,5,5],"arg_names":["self","name","ty"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo.__replace__","name":"__replace__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,5,5],"arg_names":["self","name","ty"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.ArgInfo","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__replace__ of ArgInfo","ret_type":"torch._inductor.codegen.cuda.cuda_template.ArgInfo","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"name":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_property","is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo.name","name":"name","setter_type":null,"type":"builtins.str"}},"ty":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_property","is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo.ty","name":"ty","setter_type":null,"type":"builtins.str"}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cuda_template.ArgInfo.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cuda_template.ArgInfo","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"BaseSchedulerNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.scheduler.BaseSchedulerNode","kind":"Gdef"},"Buffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Buffer","kind":"Gdef"},"CUDABenchmarkRequest":{".class":"SymbolTableNode","cross_ref":"torch._inductor.autotune_process.CUDABenchmarkRequest","kind":"Gdef"},"CUDATemplate":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.common.KernelTemplate"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate","name":"CUDATemplate","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cuda_template","mro":["torch._inductor.codegen.cuda.cuda_template.CUDATemplate","torch._inductor.codegen.common.KernelTemplate","builtins.object"],"names":{".class":"SymbolTable","__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,1],"arg_names":["self","name","input_nodes","layout","input_reorder"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,1],"arg_names":["self","name","input_nodes","layout","input_reorder"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUDATemplate","builtins.str",{".class":"Instance","args":["torch._inductor.ir.Buffer"],"extra_attrs":null,"type_ref":"builtins.list"},"torch._inductor.ir.Layout",{".class":"UnionType","items":[{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of CUDATemplate","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"generate":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,4],"arg_names":["self","description","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.generate","name":"generate","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,4],"arg_names":["self","description","kwargs"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUDATemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"generate of CUDATemplate","ret_type":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_runtime_arg_info":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.get_runtime_arg_info","name":"get_runtime_arg_info","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUDATemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_info of CUDATemplate","ret_type":{".class":"Instance","args":["torch._inductor.codegen.cuda.cuda_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_runtime_arg_values":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,4],"arg_names":["self","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.get_runtime_arg_values","name":"get_runtime_arg_values","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,4],"arg_names":["self","kwargs"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUDATemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_values of CUDATemplate","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"globals":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.globals","name":"globals","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUDATemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"globals of CUDATemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"header":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.header","name":"header","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUDATemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"header of CUDATemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"index_counter":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.index_counter","name":"index_counter","setter_type":null,"type":{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"itertools.count"}}},"input_nodes":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.input_nodes","name":"input_nodes","setter_type":null,"type":{".class":"Instance","args":["torch._inductor.ir.Buffer"],"extra_attrs":null,"type_ref":"builtins.list"}}},"input_reorder":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.input_reorder","name":"input_reorder","setter_type":null,"type":{".class":"UnionType","items":[{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false}}},"layout":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.layout","name":"layout","setter_type":null,"type":"torch._inductor.ir.Layout"}},"output_node":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.output_node","name":"output_node","setter_type":null,"type":"torch._inductor.ir.Buffer"}},"render":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,4],"arg_names":["self","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.render","name":"render","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,4],"arg_names":["self","kwargs"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUDATemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"render of CUDATemplate","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"supports_epilogue_fusion":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.supports_epilogue_fusion","name":"supports_epilogue_fusion","original_first_arg":"op","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["op"],"arg_types":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_template.GemmOperation"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"supports_epilogue_fusion of CUDATemplate","ret_type":"builtins.bool","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.supports_epilogue_fusion","name":"supports_epilogue_fusion","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0],"arg_names":["op"],"arg_types":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cuda_template.GemmOperation"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"supports_epilogue_fusion of CUDATemplate","ret_type":"builtins.bool","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cuda_template.CUDATemplate","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"CUDATemplateBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.CUDATemplateBuffer","kind":"Gdef"},"CUDATemplateCaller":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateCaller","kind":"Gdef"},"CUDATemplateKernel":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cuda_kernel.CUDATemplateKernel","kind":"Gdef"},"CUTLASSTemplate":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.cuda.cuda_template.CUDATemplate"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate","name":"CUTLASSTemplate","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cuda_template","mro":["torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate","torch._inductor.codegen.cuda.cuda_template.CUDATemplate","torch._inductor.codegen.common.KernelTemplate","builtins.object"],"names":{".class":"SymbolTable","_DTYPE_TO_CUTLASS":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate._DTYPE_TO_CUTLASS","name":"_DTYPE_TO_CUTLASS","setter_type":null,"type":{".class":"Instance","args":["torch._C.dtype","builtins.str"],"extra_attrs":null,"type_ref":"builtins.dict"}}},"_DTYPE_TO_CUTLASS_SPARSE_META":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate._DTYPE_TO_CUTLASS_SPARSE_META","name":"_DTYPE_TO_CUTLASS_SPARSE_META","setter_type":null,"type":{".class":"Instance","args":["torch._C.dtype","builtins.str"],"extra_attrs":null,"type_ref":"builtins.dict"}}},"cute_int":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","int_str","var_name"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate.cute_int","name":"cute_int","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","int_str","var_name"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"cute_int of CUTLASSTemplate","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"cutlass_sparse_meta_type_cast":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","node","ptr"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate.cutlass_sparse_meta_type_cast","name":"cutlass_sparse_meta_type_cast","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","node","ptr"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate","torch._inductor.ir.IRNode","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"cutlass_sparse_meta_type_cast of CUTLASSTemplate","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"cutlass_type_cast":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","node","ptr"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate.cutlass_type_cast","name":"cutlass_type_cast","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","node","ptr"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate","torch._inductor.ir.IRNode","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"cutlass_type_cast of CUTLASSTemplate","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_runtime_arg_info":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated","is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate.get_runtime_arg_info","name":"get_runtime_arg_info","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_info of CUTLASSTemplate","ret_type":{".class":"Instance","args":["torch._inductor.codegen.cuda.cuda_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate.get_runtime_arg_info","name":"get_runtime_arg_info","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_info of CUTLASSTemplate","ret_type":{".class":"Instance","args":["torch._inductor.codegen.cuda.cuda_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"get_runtime_arg_values":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,4],"arg_names":["self","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated","is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate.get_runtime_arg_values","name":"get_runtime_arg_values","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,4],"arg_names":["self","kwargs"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_values of CUTLASSTemplate","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate.get_runtime_arg_values","name":"get_runtime_arg_values","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,4],"arg_names":["self","kwargs"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_values of CUTLASSTemplate","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"globals":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate.globals","name":"globals","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"globals of CUTLASSTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"header":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate.header","name":"header","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"header of CUTLASSTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cuda_template.CUTLASSTemplate","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"DTYPE_TO_CUTLASS_TYPE":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cutlass_utils.DTYPE_TO_CUTLASS_TYPE","kind":"Gdef"},"GemmOperation":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeAlias","alias_tvars":[],"column":0,"fullname":"torch._inductor.codegen.cuda.cuda_template.GemmOperation","line":30,"no_args":false,"normalized":false,"python_3_12_type_alias":false,"target":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}}},"IRNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.IRNode","kind":"Gdef"},"IndentedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.IndentedBuffer","kind":"Gdef"},"KernelTemplate":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.KernelTemplate","kind":"Gdef"},"Layout":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Layout","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"Placeholder":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.Placeholder","kind":"Gdef"},"TYPE_CHECKING":{".class":"SymbolTableNode","cross_ref":"typing.TYPE_CHECKING","kind":"Gdef"},"TensorMeta":{".class":"SymbolTableNode","cross_ref":"torch._inductor.autotune_process.TensorMeta","kind":"Gdef"},"V":{".class":"SymbolTableNode","cross_ref":"torch._inductor.virtualized.V","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_template.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_template.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_template.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_template.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_template.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cuda_template.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"autotuning_log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cuda_template.autotuning_log","name":"autotuning_log","setter_type":null,"type":"logging.Logger"}},"dataclass":{".class":"SymbolTableNode","cross_ref":"dataclasses.dataclass","kind":"Gdef"},"functools":{".class":"SymbolTableNode","cross_ref":"functools","kind":"Gdef"},"getArtifactLogger":{".class":"SymbolTableNode","cross_ref":"torch._logging._internal.getArtifactLogger","kind":"Gdef"},"hashlib":{".class":"SymbolTableNode","cross_ref":"hashlib","kind":"Gdef"},"itertools":{".class":"SymbolTableNode","cross_ref":"itertools","kind":"Gdef"},"override":{".class":"SymbolTableNode","cross_ref":"typing.override","kind":"Gdef"},"patch":{".class":"SymbolTableNode","cross_ref":"unittest.mock.patch","kind":"Gdef"},"sympy":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cuda_template.sympy","name":"sympy","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cuda_template.sympy","source_any":null,"type_of_any":3}}},"torch":{".class":"SymbolTableNode","cross_ref":"torch","kind":"Gdef"},"unique":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.unique","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cuda_template.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cuda_template.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959418,"dep_lines":[21,22,20,13,16,17,19,26,8,14,2,3,4,5,6,7,12,1,1,1,1,1,1,10],"dep_prios":[5,5,5,5,5,5,5,25,5,5,10,10,10,5,5,5,10,5,30,30,30,30,30,10],"dependencies":["torch._inductor.codegen.cuda.cuda_kernel","torch._inductor.codegen.cuda.cutlass_utils","torch._inductor.codegen.common","torch._inductor.utils","torch._inductor.autotune_process","torch._inductor.ir","torch._inductor.virtualized","torch._inductor.scheduler","unittest.mock","torch._logging","functools","hashlib","itertools","dataclasses","typing","typing_extensions","torch","builtins","_frozen_importlib","abc","logging","torch._C","torch._logging._internal"],"hash":"ef109adeb3678df91c2e89fc56d84398f205ab43","id":"torch._inductor.codegen.cuda.cuda_template","ignore_all":true,"interface_hash":"4e240db191af0e6a77c1012ceb4045c47775f04d","mtime":1757363945,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cuda_template.py","plugin_data":null,"size":11526,"suppressed":["sympy"],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_cache.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda.cutlass_cache","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"CONFIG_PREFIX":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.CONFIG_PREFIX","name":"CONFIG_PREFIX","setter_type":null,"type":"builtins.str"}},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"_generate_config_filename":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["request_key"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_cache._generate_config_filename","name":"_generate_config_filename","original_first_arg":"request_key","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["request_key"],"arg_types":["builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_generate_config_filename","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"cache_dir":{".class":"SymbolTableNode","cross_ref":"torch._inductor.runtime.cache_dir_utils.cache_dir","kind":"Gdef"},"clear_on_fresh_cache":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.clear_on_fresh_cache","kind":"Gdef"},"config":{".class":"SymbolTableNode","cross_ref":"torch._inductor.config","kind":"Gdef"},"cutlass_key":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codecache.cutlass_key","kind":"Gdef"},"functools":{".class":"SymbolTableNode","cross_ref":"functools","kind":"Gdef"},"get_config_request_key":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["arch","cuda_version","instantiation_level"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.get_config_request_key","name":"get_config_request_key","original_first_arg":"arch","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["arch","cuda_version","instantiation_level"],"arg_types":["builtins.str","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_config_request_key","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_cuda_arch":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cuda_env.get_cuda_arch","kind":"Gdef"},"get_cuda_version":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cuda_env.get_cuda_version","kind":"Gdef"},"get_cutlass_operation_serializer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.serialization.get_cutlass_operation_serializer","kind":"Gdef"},"hashlib":{".class":"SymbolTableNode","cross_ref":"hashlib","kind":"Gdef"},"json":{".class":"SymbolTableNode","cross_ref":"json","kind":"Gdef"},"log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.log","name":"log","setter_type":null,"type":"logging.Logger"}},"logging":{".class":"SymbolTableNode","cross_ref":"logging","kind":"Gdef"},"maybe_fetch_ops":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[],"arg_names":[],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.maybe_fetch_ops","name":"maybe_fetch_ops","original_first_arg":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"maybe_fetch_ops","ret_type":{".class":"UnionType","items":[{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_cache.maybe_fetch_ops","name":"maybe_fetch_ops","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}}}},"os":{".class":"SymbolTableNode","cross_ref":"os","kind":"Gdef"},"time":{".class":"SymbolTableNode","cross_ref":"time","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_cache.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_cache.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959417,"dep_lines":[12,13,14,10,11,15,10,2,3,4,5,6,7,8,10,1,1,1],"dep_prios":[5,5,5,10,5,5,20,10,10,10,10,10,10,5,20,5,30,30],"dependencies":["torch._inductor.codegen.cuda.cuda_env","torch._inductor.codegen.cuda.serialization","torch._inductor.runtime.cache_dir_utils","torch._inductor.config","torch._inductor.codecache","torch._inductor.utils","torch._inductor","functools","hashlib","json","logging","os","time","typing","torch","builtins","_frozen_importlib","abc"],"hash":"5446ec665da99fd5c1b3aeeeb0eff63f5406d15e","id":"torch._inductor.codegen.cuda.cutlass_cache","ignore_all":true,"interface_hash":"099804c41522330b1dafb7e6345fb5e48eb251f3","mtime":1757363945,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_cache.py","plugin_data":null,"size":3191,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_lib_extensions/__init__.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__path__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.__path__","name":"__path__","setter_type":null,"type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"}}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_lib_extensions\\__init__.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_lib_extensions/__init__.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757957943,"dep_lines":[1,1,1,1],"dep_prios":[5,30,30,30],"dependencies":["builtins","_frozen_importlib","abc","typing"],"hash":"da39a3ee5e6b4b0d3255bfef95601890afd80709","id":"torch._inductor.codegen.cuda.cutlass_lib_extensions","ignore_all":true,"interface_hash":"9a65e4302e9c2df56bd3287016d79e24fbf7dd46","mtime":1757363945,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_lib_extensions\\__init__.py","plugin_data":null,"size":0,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_lib_extensions/evt_extensions.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"Buffer":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeAlias","alias_tvars":[],"column":0,"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Buffer","line":16,"no_args":false,"normalized":false,"python_3_12_type_alias":false,"target":{".class":"UnionType","items":["torch._inductor.ir.ComputedBuffer","torch._inductor.ir.InputBuffer"],"uses_pep604_syntax":false}}},"Callable":{".class":"SymbolTableNode","cross_ref":"typing.Callable","kind":"Gdef"},"CollectiveEpilogue":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.CollectiveEpilogue","name":"CollectiveEpilogue","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.CollectiveEpilogue","source_any":null,"type_of_any":3}}},"ComputedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.ComputedBuffer","kind":"Gdef"},"CutlassArgType":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeAlias","alias_tvars":[],"column":0,"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.CutlassArgType","line":19,"no_args":false,"normalized":false,"python_3_12_type_alias":false,"target":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}}},"CutlassTensor":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.CutlassTensor","name":"CutlassTensor","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.CutlassTensor","source_any":null,"type_of_any":3}}},"CutlassTupleType":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeAlias","alias_tvars":[],"column":0,"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.CutlassTupleType","line":17,"no_args":false,"normalized":false,"python_3_12_type_alias":false,"target":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}}},"CutlassVisitorType":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeAlias","alias_tvars":[],"column":0,"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.CutlassVisitorType","line":18,"no_args":false,"normalized":false,"python_3_12_type_alias":false,"target":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}}},"DataType":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.DataType","name":"DataType","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.DataType","source_any":null,"type_of_any":3}}},"EmptyByte":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.EmptyByte","name":"EmptyByte","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.EmptyByte","source_any":null,"type_of_any":3}}},"EpilogueFunctor":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeAlias","alias_tvars":[],"column":0,"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.EpilogueFunctor","line":15,"no_args":false,"normalized":false,"python_3_12_type_alias":false,"target":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}}},"EpilogueFunctorVisitor":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.EpilogueFunctorVisitor","name":"EpilogueFunctorVisitor","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.EpilogueFunctorVisitor","source_any":null,"type_of_any":3}}},"EpilogueScheduleType":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.EpilogueScheduleType","name":"EpilogueScheduleType","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.EpilogueScheduleType","source_any":null,"type_of_any":3}}},"Expr":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Expr","name":"Expr","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Expr","source_any":null,"type_of_any":3}}},"FusionCallbacks":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.FusionCallbacks","name":"FusionCallbacks","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.FusionCallbacks","source_any":null,"type_of_any":3}}},"IndentedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.IndentedBuffer","kind":"Gdef"},"InputBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.InputBuffer","kind":"Gdef"},"LayoutType":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.LayoutType","name":"LayoutType","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.LayoutType","source_any":null,"type_of_any":3}}},"OrderedSet":{".class":"SymbolTableNode","cross_ref":"torch.utils._ordered_set.OrderedSet","kind":"Gdef"},"PythonASTFrontend":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.PythonASTFrontend","name":"PythonASTFrontend","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.PythonASTFrontend","source_any":null,"type_of_any":3}}},"TileDescription":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.TileDescription","name":"TileDescription","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.TileDescription","source_any":null,"type_of_any":3}}},"Union":{".class":"SymbolTableNode","cross_ref":"typing.Union","kind":"Gdef"},"_CUTLASS_C_DTYPES":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions._CUTLASS_C_DTYPES","name":"_CUTLASS_C_DTYPES","setter_type":null,"type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.dtype2ctype","source_any":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.dtype2ctype","source_any":null,"type_of_any":3},"type_of_any":7}],"extra_attrs":null,"type_ref":"torch.utils._ordered_set.OrderedSet"}}},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"_get_arg_from_node":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["arg_ty","node","size_hint_fn"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions._get_arg_from_node","name":"_get_arg_from_node","original_first_arg":"arg_ty","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["arg_ty","node","size_hint_fn"],"arg_types":["builtins.type",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Buffer"},{".class":"CallableType","arg_kinds":[0],"arg_names":[null],"arg_types":[{".class":"UnionType","items":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Expr","source_any":null,"type_of_any":3},"builtins.int"],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":"builtins.int","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_get_arg_from_node","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_render_argument_type":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["epilogue_functor","name_to_buffer","size_hint_fn"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions._render_argument_type","name":"_render_argument_type","original_first_arg":"epilogue_functor","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["epilogue_functor","name_to_buffer","size_hint_fn"],"arg_types":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.EpilogueFunctor"},{".class":"Instance","args":["builtins.str",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Buffer"}],"extra_attrs":null,"type_ref":"builtins.dict"},{".class":"CallableType","arg_kinds":[0],"arg_names":[null],"arg_types":[{".class":"UnionType","items":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Expr","source_any":null,"type_of_any":3},"builtins.int"],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":"builtins.int","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_render_argument_type","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_trace":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,4],"arg_names":["fn_src","example_tensors","cc","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions._trace","name":"_trace","original_first_arg":"fn_src","type":{".class":"CallableType","arg_kinds":[0,0,0,4],"arg_names":["fn_src","example_tensors","cc","kwargs"],"arg_types":["builtins.str",{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.CutlassTensor","source_any":null,"type_of_any":3}],"extra_attrs":null,"type_ref":"builtins.dict"},"builtins.int",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_trace","ret_type":{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.EpilogueFunctor"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"ast":{".class":"SymbolTableNode","cross_ref":"ast","kind":"Gdef"},"create_example_tensors":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["var_name_to_buffer_name","name_to_buffer","size_hint_fn"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.create_example_tensors","name":"create_example_tensors","original_first_arg":"var_name_to_buffer_name","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["var_name_to_buffer_name","name_to_buffer","size_hint_fn"],"arg_types":[{".class":"Instance","args":["builtins.str","builtins.str"],"extra_attrs":null,"type_ref":"builtins.dict"},{".class":"Instance","args":["builtins.str",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Buffer"}],"extra_attrs":null,"type_ref":"builtins.dict"},{".class":"CallableType","arg_kinds":[0],"arg_names":[null],"arg_types":[{".class":"UnionType","items":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Expr","source_any":null,"type_of_any":3},"builtins.int"],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":"builtins.int","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"create_example_tensors","ret_type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.CutlassTensor","source_any":null,"type_of_any":3}],"extra_attrs":null,"type_ref":"builtins.dict"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"ctypes":{".class":"SymbolTableNode","cross_ref":"ctypes","kind":"Gdef"},"cuda_env":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cuda_env","kind":"Gdef"},"dtype2ctype":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.dtype2ctype","name":"dtype2ctype","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.dtype2ctype","source_any":null,"type_of_any":3}}},"is_contiguous_strides_for_shape":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.is_contiguous_strides_for_shape","kind":"Gdef"},"textwrap":{".class":"SymbolTableNode","cross_ref":"textwrap","kind":"Gdef"},"torch_dtype_to_cutlass_type":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cutlass_utils.torch_dtype_to_cutlass_type","kind":"Gdef"},"trace":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,0,0,0,0,4],"arg_names":["fn_src","example_tensors","accum_type","output_type","tile_description","epilogue_schedule","name_to_buffer","size_hint_fn","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.trace","name":"trace","original_first_arg":"fn_src","type":{".class":"CallableType","arg_kinds":[0,0,0,0,0,0,0,0,4],"arg_names":["fn_src","example_tensors","accum_type","output_type","tile_description","epilogue_schedule","name_to_buffer","size_hint_fn","kwargs"],"arg_types":["builtins.str",{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.CutlassTensor","source_any":null,"type_of_any":3}],"extra_attrs":null,"type_ref":"builtins.dict"},{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.DataType","source_any":null,"type_of_any":3},{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.DataType","source_any":null,"type_of_any":3},{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.TileDescription","source_any":null,"type_of_any":3},{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.EpilogueScheduleType","source_any":null,"type_of_any":3},{".class":"Instance","args":["builtins.str",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Buffer"}],"extra_attrs":null,"type_ref":"builtins.dict"},{".class":"CallableType","arg_kinds":[0],"arg_names":[null],"arg_types":[{".class":"UnionType","items":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions.Expr","source_any":null,"type_of_any":3},"builtins.int"],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":"builtins.int","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]},{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.dict"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"trace","ret_type":{".class":"TupleType","implicit":false,"items":["builtins.str","builtins.str","builtins.str"],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"try_import_cutlass":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cutlass_utils.try_import_cutlass","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_lib_extensions\\evt_extensions.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_lib_extensions/evt_extensions.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959415,"dep_lines":[12,58,58,5,10,59,1,25,26,27,1,1,1,1,1,1,39,42,48,45,30,33,36,3,51],"dep_prios":[5,10,20,5,5,5,5,10,10,10,5,30,30,30,30,30,5,5,5,5,5,5,5,5,5],"dependencies":["torch._inductor.codegen.cuda.cutlass_utils","torch._inductor.codegen.cuda.cuda_env","torch._inductor.codegen.cuda","torch._inductor.ir","torch.utils._ordered_set","torch._inductor.utils","typing","ast","ctypes","textwrap","builtins","_frozen_importlib","abc","functools","torch._inductor.codegen.common","torch.utils"],"hash":"a53a75aa7b742af73f48c50cc4644bab356f9c5e","id":"torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions","ignore_all":true,"interface_hash":"38e26482f6975e15618a5ee5f59fbb667f3f594e","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_lib_extensions\\evt_extensions.py","plugin_data":null,"size":10083,"suppressed":["cutlass.backend.evt.backend.emitter_base","cutlass.backend.evt.backend.sm90_emitter","cutlass.backend.evt.ir.tensor","cutlass.backend.evt.frontend","cutlass.backend.c_types","cutlass.backend.epilogue","cutlass.backend.evt","sympy","cutlass_library"],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_lib_extensions/gemm_operation_extensions.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","EmitGemmUniversal3xInstanceWithEVT":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT","name":"EmitGemmUniversal3xInstanceWithEVT","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions","mro":["torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT","builtins.object"],"names":{".class":"SymbolTable","__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,1,1],"arg_names":["self","operation_suffix","evt_name"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.__init__","name":"__init__","original_first_arg":"self","type":null}},"builtin_epilogue_functor_template":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.builtin_epilogue_functor_template","name":"builtin_epilogue_functor_template","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}}},"emit":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","operation"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.emit","name":"emit","original_first_arg":"self","type":null}},"emit_block_scale_epilogue_functor":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","operation"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.emit_block_scale_epilogue_functor","name":"emit_block_scale_epilogue_functor","original_first_arg":"self","type":null}},"evt_name":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.evt_name","name":"evt_name","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}}},"gemm_template":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.gemm_template","name":"gemm_template","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}}},"includes":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.includes","name":"includes","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}}},"instance_template":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.instance_template","name":"instance_template","original_first_arg":"self","type":null}},"operation_suffix":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.operation_suffix","name":"operation_suffix","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}}},"pointerize_if_grouped":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["operation","layout"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.pointerize_if_grouped","name":"pointerize_if_grouped","original_first_arg":"operation","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.pointerize_if_grouped","name":"pointerize_if_grouped","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["operation","layout"],"arg_types":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"pointerize_if_grouped of EmitGemmUniversal3xInstanceWithEVT","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"problem_shape":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["operation"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.problem_shape","name":"problem_shape","original_first_arg":"operation","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.problem_shape","name":"problem_shape","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0],"arg_names":["operation"],"arg_types":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"problem_shape of EmitGemmUniversal3xInstanceWithEVT","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.EmitGemmUniversal3xInstanceWithEVT","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"_LOGGER":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions._LOGGER","name":"_LOGGER","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":5},"type_of_any":7}}},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"enum":{".class":"SymbolTableNode","cross_ref":"enum","kind":"Gdef"},"try_import_cutlass":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cutlass_utils.try_import_cutlass","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_lib_extensions\\gemm_operation_extensions.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_lib_extensions/gemm_operation_extensions.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959415,"dep_lines":[2,9,1,1,1,1,1,11,12],"dep_prios":[5,10,5,30,30,30,30,5,5],"dependencies":["torch._inductor.codegen.cuda.cutlass_utils","enum","builtins","_frozen_importlib","abc","functools","typing"],"hash":"531eb815be73bbe650cd51ce26b6bbc52c4f35a0","id":"torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions","ignore_all":true,"interface_hash":"29028375d50515b322468bdba11cda6ee0ee6306","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_lib_extensions\\gemm_operation_extensions.py","plugin_data":null,"size":19057,"suppressed":["cutlass_library.gemm_operation","cutlass_library.library"],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_presets.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda.cutlass_presets","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_presets.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_presets.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_presets.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_presets.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_presets.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_presets.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"defaultdict":{".class":"SymbolTableNode","cross_ref":"collections.defaultdict","kind":"Gdef"},"functools":{".class":"SymbolTableNode","cross_ref":"functools","kind":"Gdef"},"gen_cutlass_presets":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[],"arg_names":[],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_presets.gen_cutlass_presets","name":"gen_cutlass_presets","original_first_arg":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"gen_cutlass_presets","ret_type":{".class":"Instance","args":["builtins.int",{".class":"Instance","args":["builtins.str",{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"}],"extra_attrs":null,"type_ref":"builtins.dict"}],"extra_attrs":null,"type_ref":"builtins.dict"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_presets.gen_cutlass_presets","name":"gen_cutlass_presets","setter_type":null,"type":{".class":"Instance","args":[{".class":"Instance","args":["builtins.int",{".class":"Instance","args":["builtins.str",{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"}],"extra_attrs":null,"type_ref":"builtins.dict"}],"extra_attrs":null,"type_ref":"builtins.dict"}],"extra_attrs":null,"type_ref":"functools._lru_cache_wrapper"}}}},"get_cuda_arch":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cuda_env.get_cuda_arch","kind":"Gdef"},"torch":{".class":"SymbolTableNode","cross_ref":"torch","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_presets.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_presets.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959353,"dep_lines":[5,1,2,4,1,1,1,1],"dep_prios":[5,10,5,10,5,30,30,30],"dependencies":["torch._inductor.codegen.cuda.cuda_env","functools","collections","torch","builtins","_frozen_importlib","abc","typing"],"hash":"041570ab18e1be7304e5b79fd9f7fedafaff9a56","id":"torch._inductor.codegen.cuda.cutlass_presets","ignore_all":true,"interface_hash":"4f9c4f848eb4239cb6e8149768856bb75d8fa4e8","mtime":1757363945,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_presets.py","plugin_data":null,"size":25745,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_python_evt.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda.cutlass_python_evt","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"BaseSchedulerNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.scheduler.BaseSchedulerNode","kind":"Gdef"},"ComputedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.ComputedBuffer","kind":"Gdef"},"CutlassEVTCodegen":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen","name":"CutlassEVTCodegen","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cutlass_python_evt","mro":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen","torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn","builtins.object"],"names":{".class":"SymbolTable","D_var_name":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.D_var_name","name":"D_var_name","setter_type":null,"type":{".class":"NoneType"}}},"__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","accumulator_node_name","removed_buffers"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","accumulator_node_name","removed_buffers"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen","builtins.str",{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"torch.utils._ordered_set.OrderedSet"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of CutlassEVTCodegen","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_check_indexing":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","name","index"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen._check_indexing","name":"_check_indexing","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","name","index"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen","builtins.str",{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_python_evt.sympy","source_any":null,"type_of_any":3}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_check_indexing of CutlassEVTCodegen","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_get_cur_node":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen._get_cur_node","name":"_get_cur_node","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_get_cur_node of CutlassEVTCodegen","ret_type":"torch._inductor.ir.ComputedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_get_current_index_vars":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen._get_current_index_vars","name":"_get_current_index_vars","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_get_current_index_vars of CutlassEVTCodegen","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_python_evt.sympy","source_any":null,"type_of_any":3}],"extra_attrs":null,"type_ref":"typing.Sequence"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_render_input_signature":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen._render_input_signature","name":"_render_input_signature","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_render_input_signature of CutlassEVTCodegen","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_render_return_statement":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen._render_return_statement","name":"_render_return_statement","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_render_return_statement of CutlassEVTCodegen","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_stride_compatible":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","left","right"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen._stride_compatible","name":"_stride_compatible","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","left","right"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen",{".class":"Instance","args":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_python_evt.sympy","source_any":null,"type_of_any":3}],"extra_attrs":null,"type_ref":"typing.Iterable"},{".class":"Instance","args":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_python_evt.sympy","source_any":null,"type_of_any":3}],"extra_attrs":null,"type_ref":"typing.Iterable"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_stride_compatible of CutlassEVTCodegen","ret_type":"builtins.bool","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_tmp_var":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen._tmp_var","name":"_tmp_var","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_tmp_var of CutlassEVTCodegen","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"accumulator_node_name":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.accumulator_node_name","name":"accumulator_node_name","setter_type":null,"type":"builtins.str"}},"body":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.body","name":"body","setter_type":null,"type":"torch._inductor.utils.IndentedBuffer"}},"cur_node":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.cur_node","name":"cur_node","setter_type":null,"type":{".class":"UnionType","items":["torch._inductor.ir.ComputedBuffer",{".class":"NoneType"}],"uses_pep604_syntax":false}}},"finalize":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.finalize","name":"finalize","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"finalize of CutlassEVTCodegen","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_index_vars":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["node"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.get_index_vars","name":"get_index_vars","original_first_arg":"node","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["node"],"arg_types":["torch._inductor.ir.ComputedBuffer"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_index_vars of CutlassEVTCodegen","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_python_evt.sympy","source_any":null,"type_of_any":3}],"extra_attrs":null,"type_ref":"typing.Sequence"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.get_index_vars","name":"get_index_vars","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0],"arg_names":["node"],"arg_types":["torch._inductor.ir.ComputedBuffer"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_index_vars of CutlassEVTCodegen","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_python_evt.sympy","source_any":null,"type_of_any":3}],"extra_attrs":null,"type_ref":"typing.Sequence"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"get_reads":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.get_reads","name":"get_reads","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_reads of CutlassEVTCodegen","ret_type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_renames":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.get_renames","name":"get_renames","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_renames of CutlassEVTCodegen","ret_type":{".class":"Instance","args":["builtins.str","builtins.str"],"extra_attrs":null,"type_ref":"builtins.dict"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_value":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.get_value","name":"get_value","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_value of CutlassEVTCodegen","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_writes":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.get_writes","name":"get_writes","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_writes of CutlassEVTCodegen","ret_type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"ir_to_evt_python_code":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["cuda_template_node_name","epilogue_nodes","removed_buffers"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.ir_to_evt_python_code","name":"ir_to_evt_python_code","original_first_arg":"cuda_template_node_name","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["cuda_template_node_name","epilogue_nodes","removed_buffers"],"arg_types":["builtins.str",{".class":"Instance","args":["torch._inductor.scheduler.BaseSchedulerNode"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"torch.utils._ordered_set.OrderedSet"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"ir_to_evt_python_code of CutlassEVTCodegen","ret_type":{".class":"TupleType","implicit":false,"items":[{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.dict"},"builtins.str"],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.ir_to_evt_python_code","name":"ir_to_evt_python_code","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["cuda_template_node_name","epilogue_nodes","removed_buffers"],"arg_types":["builtins.str",{".class":"Instance","args":["torch._inductor.scheduler.BaseSchedulerNode"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"torch.utils._ordered_set.OrderedSet"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"ir_to_evt_python_code of CutlassEVTCodegen","ret_type":{".class":"TupleType","implicit":false,"items":[{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.dict"},"builtins.str"],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"is_D_assigned":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.is_D_assigned","name":"is_D_assigned","setter_type":null,"type":"builtins.bool"}},"last_stored_var_name":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.last_stored_var_name","name":"last_stored_var_name","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},"type_of_any":7}}},"load":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","name","index"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.load","name":"load","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","name","index"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen","builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"load of CutlassEVTCodegen","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"name_to_buffer":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.name_to_buffer","name":"name_to_buffer","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"UnionType","items":["torch._inductor.ir.Buffer","torch._inductor.ir.TensorBox","torch._inductor.ir.TorchBindObject",{".class":"AnyType","missing_import_name":"torch._inductor.graph.sympy","source_any":null,"type_of_any":3}],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"reads":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.reads","name":"reads","setter_type":null,"type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"torch.utils._ordered_set.OrderedSet"}}},"removed_buffers":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.removed_buffers","name":"removed_buffers","setter_type":null,"type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"torch.utils._ordered_set.OrderedSet"}}},"set_cur_node":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","node"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_generator","is_decorated","is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.set_cur_node","name":"set_cur_node","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","node"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen","torch._inductor.ir.ComputedBuffer"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"set_cur_node of CutlassEVTCodegen","ret_type":{".class":"Instance","args":[{".class":"NoneType"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"typing.Generator"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.set_cur_node","name":"set_cur_node","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","node"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen","torch._inductor.ir.ComputedBuffer"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"set_cur_node of CutlassEVTCodegen","ret_type":{".class":"Instance","args":[{".class":"NoneType"},{".class":"NoneType"},{".class":"NoneType"}],"extra_attrs":null,"type_ref":"contextlib._GeneratorContextManager"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"store":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,1,1,1],"arg_names":["self","name","index","value","mode"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.store","name":"store","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,1,1,1],"arg_names":["self","name","index","value","mode"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"store of CutlassEVTCodegen","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"store_name_to_value":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.store_name_to_value","name":"store_name_to_value","setter_type":null,"type":{".class":"Instance","args":["builtins.str","torch._inductor.virtualized.OpsValue"],"extra_attrs":null,"type_ref":"builtins.dict"}}},"var_counter":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.var_counter","name":"var_counter","setter_type":null,"type":{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"typing.Iterator"}}},"var_name_to_buffer_name":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.var_name_to_buffer_name","name":"var_name_to_buffer_name","setter_type":null,"type":{".class":"Instance","args":["builtins.str","builtins.str"],"extra_attrs":null,"type_ref":"builtins.dict"}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"CutlassEVTOpsMixIn":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn","name":"CutlassEVTOpsMixIn","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cutlass_python_evt","mro":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn","builtins.object"],"names":{".class":"SymbolTable","_infix_bin_op":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["op","a","b"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn._infix_bin_op","name":"_infix_bin_op","original_first_arg":"op","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["op","a","b"],"arg_types":["builtins.str","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_infix_bin_op of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn._infix_bin_op","name":"_infix_bin_op","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["op","a","b"],"arg_types":["builtins.str","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_infix_bin_op of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"_prefix_bin_op":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["op","a","b"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn._prefix_bin_op","name":"_prefix_bin_op","original_first_arg":"op","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["op","a","b"],"arg_types":["builtins.str","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_prefix_bin_op of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn._prefix_bin_op","name":"_prefix_bin_op","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["op","a","b"],"arg_types":["builtins.str","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_prefix_bin_op of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"_prefix_un_op":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["op","a"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn._prefix_un_op","name":"_prefix_un_op","original_first_arg":"op","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["op","a"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_prefix_un_op of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn._prefix_un_op","name":"_prefix_un_op","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["op","a"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_prefix_un_op of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"add":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["x0","x1"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.add","name":"add","original_first_arg":"x0","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["x0","x1"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"add of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.add","name":"add","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["x0","x1"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"add of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"constant":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["value","dtype"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.constant","name":"constant","original_first_arg":"value","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["value","dtype"],"arg_types":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"constant of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.constant","name":"constant","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["value","dtype"],"arg_types":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"constant of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"ge":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["x0","x1"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.ge","name":"ge","original_first_arg":"x0","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["x0","x1"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"ge of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.ge","name":"ge","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["x0","x1"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"ge of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"mul":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["x0","x1"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.mul","name":"mul","original_first_arg":"x0","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["x0","x1"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"mul of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.mul","name":"mul","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["x0","x1"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"mul of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"relu":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["x0"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.relu","name":"relu","original_first_arg":"x0","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["x0"],"arg_types":["builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"relu of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.relu","name":"relu","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0],"arg_names":["x0"],"arg_types":["builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"relu of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"sigmoid":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["x0"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.sigmoid","name":"sigmoid","original_first_arg":"x0","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["x0"],"arg_types":["builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"sigmoid of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.sigmoid","name":"sigmoid","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0],"arg_names":["x0"],"arg_types":["builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"sigmoid of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"sub":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["x0","x1"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.sub","name":"sub","original_first_arg":"x0","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["x0","x1"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"sub of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.sub","name":"sub","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["x0","x1"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"sub of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"tanh":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["x0"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.tanh","name":"tanh","original_first_arg":"x0","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["x0"],"arg_types":["builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"tanh of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.tanh","name":"tanh","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0],"arg_names":["x0"],"arg_types":["builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"tanh of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"to_dtype":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,1,1],"arg_names":["x","dtype","src_dtype","use_compute_types"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.to_dtype","name":"to_dtype","original_first_arg":"x","type":{".class":"CallableType","arg_kinds":[0,0,1,1],"arg_names":["x","dtype","src_dtype","use_compute_types"],"arg_types":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"UnionType","items":["torch._C.dtype",{".class":"NoneType"}],"uses_pep604_syntax":false},"builtins.bool"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"to_dtype of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.to_dtype","name":"to_dtype","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0,1,1],"arg_names":["x","dtype","src_dtype","use_compute_types"],"arg_types":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"UnionType","items":["torch._C.dtype",{".class":"NoneType"}],"uses_pep604_syntax":false},"builtins.bool"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"to_dtype of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"truediv":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["x0","x1"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.truediv","name":"truediv","original_first_arg":"x0","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["x0","x1"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"truediv of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.truediv","name":"truediv","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["x0","x1"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"truediv of CutlassEVTOpsMixIn","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"DefaultHandler":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ops_handler.DefaultHandler","kind":"Gdef"},"DelayReplaceLine":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.DelayReplaceLine","kind":"Gdef"},"Generator":{".class":"SymbolTableNode","cross_ref":"typing.Generator","kind":"Gdef"},"IndentedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.IndentedBuffer","kind":"Gdef"},"Iterable":{".class":"SymbolTableNode","cross_ref":"typing.Iterable","kind":"Gdef"},"Iterator":{".class":"SymbolTableNode","cross_ref":"typing.Iterator","kind":"Gdef"},"MockCutlassHandler":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn","torch._inductor.ops_handler.WrapperHandler"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.MockCutlassHandler","name":"MockCutlassHandler","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.MockCutlassHandler","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cutlass_python_evt","mro":["torch._inductor.codegen.cuda.cutlass_python_evt.MockCutlassHandler","torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTOpsMixIn","torch._inductor.ops_handler.WrapperHandler","torch._inductor.ops_handler.DefaultHandler","torch._inductor.ops_handler.OpsHandler","builtins.object"],"names":{".class":"SymbolTable"},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.MockCutlassHandler.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cutlass_python_evt.MockCutlassHandler","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"OpsValue":{".class":"SymbolTableNode","cross_ref":"torch._inductor.virtualized.OpsValue","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"OrderedSet":{".class":"SymbolTableNode","cross_ref":"torch.utils._ordered_set.OrderedSet","kind":"Gdef"},"Pointwise":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Pointwise","kind":"Gdef"},"Sequence":{".class":"SymbolTableNode","cross_ref":"typing.Sequence","kind":"Gdef"},"V":{".class":"SymbolTableNode","cross_ref":"torch._inductor.virtualized.V","kind":"Gdef"},"WrapperHandler":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ops_handler.WrapperHandler","kind":"Gdef"},"_ACCUMULATOR_ARG_NAME":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt._ACCUMULATOR_ARG_NAME","name":"_ACCUMULATOR_ARG_NAME","setter_type":null,"type":"builtins.str"}},"_AssignmentFormatter":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.ops_handler.DefaultHandler"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cutlass_python_evt._AssignmentFormatter","name":"_AssignmentFormatter","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt._AssignmentFormatter","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cutlass_python_evt","mro":["torch._inductor.codegen.cuda.cutlass_python_evt._AssignmentFormatter","torch._inductor.ops_handler.DefaultHandler","torch._inductor.ops_handler.OpsHandler","builtins.object"],"names":{".class":"SymbolTable","__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","parent_handler"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt._AssignmentFormatter.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","parent_handler"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt._AssignmentFormatter","torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of _AssignmentFormatter","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_default":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0],"arg_names":["self","name","args","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt._AssignmentFormatter._default","name":"_default","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0],"arg_names":["self","name","args","kwargs"],"arg_types":["torch._inductor.codegen.cuda.cutlass_python_evt._AssignmentFormatter","builtins.str",{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.tuple"},{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.dict"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_default of _AssignmentFormatter","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"parent_handler":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt._AssignmentFormatter.parent_handler","name":"parent_handler","setter_type":null,"type":"torch._inductor.codegen.cuda.cutlass_python_evt.CutlassEVTCodegen"}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt._AssignmentFormatter.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cutlass_python_evt._AssignmentFormatter","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"contextmanager":{".class":"SymbolTableNode","cross_ref":"contextlib.contextmanager","kind":"Gdef"},"itertools":{".class":"SymbolTableNode","cross_ref":"itertools","kind":"Gdef"},"linesep":{".class":"SymbolTableNode","cross_ref":"os.linesep","kind":"Gdef"},"scaled_mm_evt":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0],"arg_names":["scale_A_name","scale_B_name","bias_name","output_name"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.scaled_mm_evt","name":"scaled_mm_evt","original_first_arg":"scale_A_name","type":{".class":"CallableType","arg_kinds":[0,0,0,0],"arg_names":["scale_A_name","scale_B_name","bias_name","output_name"],"arg_types":["builtins.str","builtins.str",{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},"builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"scaled_mm_evt","ret_type":{".class":"TupleType","implicit":false,"items":[{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.dict"},"builtins.str"],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"sympy":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_python_evt.sympy","name":"sympy","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_python_evt.sympy","source_any":null,"type_of_any":3}}},"torch":{".class":"SymbolTableNode","cross_ref":"torch","kind":"Gdef"},"virtualized":{".class":"SymbolTableNode","cross_ref":"torch._inductor.virtualized","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_python_evt.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_python_evt.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959410,"dep_lines":[10,11,12,13,14,2,10,1,3,4,5,9,1,1,1,1,1,1,1,1,1,1,1,1,7],"dep_prios":[5,5,5,5,5,5,20,10,5,5,5,10,5,30,30,30,30,30,30,30,30,30,30,30,10],"dependencies":["torch._inductor.virtualized","torch._inductor.ir","torch._inductor.ops_handler","torch._inductor.scheduler","torch._inductor.utils","collections.abc","torch._inductor","itertools","contextlib","os","typing","torch","builtins","_collections_abc","_frozen_importlib","abc","torch._C","torch._inductor.codegen.common","torch._inductor.graph","torch._tensor","torch.fx","torch.fx.interpreter","torch.utils","torch.utils._ordered_set"],"hash":"9b942b6dd3d3cdde12749cf4ff075dc221ad377d","id":"torch._inductor.codegen.cuda.cutlass_python_evt","ignore_all":true,"interface_hash":"d53d6b660b4fae140ee92a2a29c91279dedf297f","mtime":1757363945,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_python_evt.py","plugin_data":null,"size":12056,"suppressed":["sympy"],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_utils.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda.cutlass_utils","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"CUDACompileSourceCapturingContext":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUDACompileSourceCapturingContext","name":"CUDACompileSourceCapturingContext","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUDACompileSourceCapturingContext","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.cutlass_utils","mro":["torch._inductor.codegen.cuda.cutlass_utils.CUDACompileSourceCapturingContext","builtins.object"],"names":{".class":"SymbolTable","__enter__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,2,4],"arg_names":[null,null,null],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUDACompileSourceCapturingContext.__enter__","name":"__enter__","original_first_arg":"self","type":null}},"__exit__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,2,4],"arg_names":[null,null,null],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUDACompileSourceCapturingContext.__exit__","name":"__exit__","original_first_arg":"self","type":null}},"__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUDACompileSourceCapturingContext.__init__","name":"__init__","original_first_arg":"self","type":null}},"_compile_patch":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUDACompileSourceCapturingContext._compile_patch","name":"_compile_patch","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}}},"sources":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUDACompileSourceCapturingContext.sources","name":"sources","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUDACompileSourceCapturingContext.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cutlass_utils.CUDACompileSourceCapturingContext","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"CUTLASSArgs":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs","name":"CUTLASSArgs","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs","has_param_spec_type":false,"metaclass_type":null,"metadata":{"dataclass":{"attributes":[{"alias":null,"column":4,"has_default":true,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":227,"name":"architectures","type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false}},{"alias":null,"column":4,"has_default":true,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":228,"name":"cuda_version","type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false}},{"alias":null,"column":4,"has_default":true,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":229,"name":"instantiation_level","type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false}},{"alias":null,"column":4,"has_default":true,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":230,"name":"operations","type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false}},{"alias":null,"column":4,"has_default":true,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":239,"name":"kernel_filter_file","type":{".class":"NoneType"}},{"alias":null,"column":4,"has_default":true,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":240,"name":"selected_kernel_list","type":{".class":"NoneType"}},{"alias":null,"column":4,"has_default":true,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":241,"name":"interface_dir","type":{".class":"NoneType"}}],"frozen":false},"dataclass_tag":{}},"module_name":"torch._inductor.codegen.cuda.cutlass_utils","mro":["torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs","builtins.object"],"names":{".class":"SymbolTable","__dataclass_fields__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_classvar","is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.__dataclass_fields__","name":"__dataclass_fields__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"dataclasses.Field"}],"extra_attrs":null,"type_ref":"builtins.dict"}},"plugin_generated":true},"__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,1,1,1,1,1,1,1],"arg_names":["self","architectures","cuda_version","instantiation_level","operations","kernel_filter_file","selected_kernel_list","interface_dir"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,1,1,1,1,1,1,1],"arg_names":["self","architectures","cuda_version","instantiation_level","operations","kernel_filter_file","selected_kernel_list","interface_dir"],"arg_types":["torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs",{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"NoneType"},{".class":"NoneType"},{".class":"NoneType"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of CUTLASSArgs","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"__match_args__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_ready","allow_incompatible_override"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.__match_args__","name":"__match_args__","setter_type":null,"type":{".class":"TupleType","implicit":false,"items":[{".class":"LiteralType","fallback":"builtins.str","value":"architectures"},{".class":"LiteralType","fallback":"builtins.str","value":"cuda_version"},{".class":"LiteralType","fallback":"builtins.str","value":"instantiation_level"},{".class":"LiteralType","fallback":"builtins.str","value":"operations"},{".class":"LiteralType","fallback":"builtins.str","value":"kernel_filter_file"},{".class":"LiteralType","fallback":"builtins.str","value":"selected_kernel_list"},{".class":"LiteralType","fallback":"builtins.str","value":"interface_dir"}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}}},"plugin_generated":true},"__mypy-post_init":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.__mypy-post_init","name":"__mypy-post_init","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-post_init of CUTLASSArgs","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"__mypy-replace":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[5,5,5,5,5,5,5],"arg_names":["architectures","cuda_version","instantiation_level","operations","kernel_filter_file","selected_kernel_list","interface_dir"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.__mypy-replace","name":"__mypy-replace","original_first_arg":"architectures","type":{".class":"CallableType","arg_kinds":[5,5,5,5,5,5,5],"arg_names":["architectures","cuda_version","instantiation_level","operations","kernel_filter_file","selected_kernel_list","interface_dir"],"arg_types":[{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"NoneType"},{".class":"NoneType"},{".class":"NoneType"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-replace of CUTLASSArgs","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_staticmethod","is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.__mypy-replace","name":"__mypy-replace","setter_type":null,"type":{".class":"CallableType","arg_kinds":[5,5,5,5,5,5,5],"arg_names":["architectures","cuda_version","instantiation_level","operations","kernel_filter_file","selected_kernel_list","interface_dir"],"arg_types":[{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"NoneType"},{".class":"NoneType"},{".class":"NoneType"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-replace of CUTLASSArgs","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"plugin_generated":true},"__post_init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.__post_init__","name":"__post_init__","original_first_arg":"self","type":null}},"__replace__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,5,5,5,5,5,5,5],"arg_names":["self","architectures","cuda_version","instantiation_level","operations","kernel_filter_file","selected_kernel_list","interface_dir"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.__replace__","name":"__replace__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,5,5,5,5,5,5,5],"arg_names":["self","architectures","cuda_version","instantiation_level","operations","kernel_filter_file","selected_kernel_list","interface_dir"],"arg_types":["torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs",{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"NoneType"},{".class":"NoneType"},{".class":"NoneType"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__replace__ of CUTLASSArgs","ret_type":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"architectures":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.architectures","name":"architectures","setter_type":null,"type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false}}},"build_dir":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.build_dir","name":"build_dir","setter_type":null,"type":"builtins.str"}},"cuda_version":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.cuda_version","name":"cuda_version","setter_type":null,"type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false}}},"curr_build_dir":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.curr_build_dir","name":"curr_build_dir","setter_type":null,"type":"builtins.str"}},"disable_full_archs_compilation":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.disable_full_archs_compilation","name":"disable_full_archs_compilation","setter_type":null,"type":"builtins.bool"}},"exclude_kernels":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.exclude_kernels","name":"exclude_kernels","setter_type":null,"type":"builtins.str"}},"filter_by_cc":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.filter_by_cc","name":"filter_by_cc","setter_type":null,"type":"builtins.bool"}},"generator_target":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.generator_target","name":"generator_target","setter_type":null,"type":"builtins.str"}},"ignore_kernels":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.ignore_kernels","name":"ignore_kernels","setter_type":null,"type":"builtins.str"}},"instantiation_level":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.instantiation_level","name":"instantiation_level","setter_type":null,"type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false}}},"interface_dir":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.interface_dir","name":"interface_dir","setter_type":null,"type":{".class":"NoneType"}}},"kernel_filter_file":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.kernel_filter_file","name":"kernel_filter_file","setter_type":null,"type":{".class":"NoneType"}}},"kernels":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.kernels","name":"kernels","setter_type":null,"type":"builtins.str"}},"operations":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.operations","name":"operations","setter_type":null,"type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false}}},"selected_kernel_list":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.selected_kernel_list","name":"selected_kernel_list","setter_type":null,"type":{".class":"NoneType"}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASSArgs","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"CUTLASS_OPERATION_KIND":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.CUTLASS_OPERATION_KIND","name":"CUTLASS_OPERATION_KIND","setter_type":null,"type":"builtins.str"}},"DTYPE_TO_CPP":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cpp_utils.DTYPE_TO_CPP","kind":"Gdef"},"DTYPE_TO_CUTLASS_TYPE":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.DTYPE_TO_CUTLASS_TYPE","name":"DTYPE_TO_CUTLASS_TYPE","setter_type":null,"type":{".class":"Instance","args":["torch._C.dtype","builtins.str"],"extra_attrs":null,"type_ref":"builtins.dict"}}},"Layout":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Layout","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"Path":{".class":"SymbolTableNode","cross_ref":"pathlib.Path","kind":"Gdef"},"V":{".class":"SymbolTableNode","cross_ref":"torch._inductor.virtualized.V","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"_gen_ops_cached":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["arch","version"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils._gen_ops_cached","name":"_gen_ops_cached","original_first_arg":"arch","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["arch","version"],"arg_types":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_gen_ops_cached","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.dict"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils._gen_ops_cached","name":"_gen_ops_cached","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}}}},"_normalize_cuda_arch":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["arch"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils._normalize_cuda_arch","name":"_normalize_cuda_arch","original_first_arg":"arch","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["arch"],"arg_types":["builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_normalize_cuda_arch","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils._normalize_cuda_arch","name":"_normalize_cuda_arch","setter_type":null,"type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"functools._lru_cache_wrapper"}}}},"_rename_cutlass_import":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["content","cutlass_modules"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils._rename_cutlass_import","name":"_rename_cutlass_import","original_first_arg":"content","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["content","cutlass_modules"],"arg_types":["builtins.str",{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_rename_cutlass_import","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"atexit":{".class":"SymbolTableNode","cross_ref":"atexit","kind":"Gdef"},"cache_dir":{".class":"SymbolTableNode","cross_ref":"torch._inductor.runtime.cache_dir_utils.cache_dir","kind":"Gdef"},"clear_on_fresh_cache":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.clear_on_fresh_cache","kind":"Gdef"},"config":{".class":"SymbolTableNode","cross_ref":"torch._inductor.config","kind":"Gdef"},"cuda_standalone_runner_compile_command":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["srcpath","exepath"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.cuda_standalone_runner_compile_command","name":"cuda_standalone_runner_compile_command","original_first_arg":"srcpath","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["srcpath","exepath"],"arg_types":["pathlib.Path","pathlib.Path"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"cuda_standalone_runner_compile_command","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"dataclass":{".class":"SymbolTableNode","cross_ref":"dataclasses.dataclass","kind":"Gdef"},"dtype_match":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["torch_dtype","cutlass_dtype"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.dtype_match","name":"dtype_match","original_first_arg":"torch_dtype","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["torch_dtype","cutlass_dtype"],"arg_types":[{".class":"UnionType","items":["torch._C.dtype",{".class":"NoneType"}],"uses_pep604_syntax":false},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"dtype_match","ret_type":"builtins.bool","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.dtype_match","name":"dtype_match","setter_type":null,"type":{".class":"Instance","args":["builtins.bool"],"extra_attrs":null,"type_ref":"functools._lru_cache_wrapper"}}}},"functools":{".class":"SymbolTableNode","cross_ref":"functools","kind":"Gdef"},"gen_ops":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[],"arg_names":[],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.gen_ops","name":"gen_ops","original_first_arg":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"gen_ops","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.dict"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_accumulator_dtype":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["input_torch_dtypes"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.get_accumulator_dtype","name":"get_accumulator_dtype","original_first_arg":"input_torch_dtypes","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["input_torch_dtypes"],"arg_types":[{".class":"Instance","args":["torch._C.dtype"],"extra_attrs":null,"type_ref":"builtins.list"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_accumulator_dtype","ret_type":{".class":"UnionType","items":["torch._C.dtype",{".class":"NoneType"}],"uses_pep604_syntax":false},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_alignments":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["torch_dtype"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.get_alignments","name":"get_alignments","original_first_arg":"torch_dtype","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["torch_dtype"],"arg_types":["torch._C.dtype"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_alignments","ret_type":{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.get_alignments","name":"get_alignments","setter_type":null,"type":{".class":"Instance","args":[{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"builtins.list"}],"extra_attrs":null,"type_ref":"functools._lru_cache_wrapper"}}}},"get_cuda_arch":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cuda_env.get_cuda_arch","kind":"Gdef"},"get_cuda_version":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cuda_env.get_cuda_version","kind":"Gdef"},"get_max_alignment":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["inductor_layout"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.get_max_alignment","name":"get_max_alignment","original_first_arg":"inductor_layout","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["inductor_layout"],"arg_types":["torch._inductor.ir.Layout"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_max_alignment","ret_type":"builtins.int","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.log","name":"log","setter_type":null,"type":"logging.Logger"}},"logging":{".class":"SymbolTableNode","cross_ref":"logging","kind":"Gdef"},"move_cutlass_compiled_cache":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[],"arg_names":[],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.move_cutlass_compiled_cache","name":"move_cutlass_compiled_cache","original_first_arg":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"move_cutlass_compiled_cache","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.move_cutlass_compiled_cache","name":"move_cutlass_compiled_cache","setter_type":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"move_cutlass_compiled_cache","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"os":{".class":"SymbolTableNode","cross_ref":"os","kind":"Gdef"},"shutil":{".class":"SymbolTableNode","cross_ref":"shutil","kind":"Gdef"},"sympy":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.sympy","name":"sympy","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.cuda.cutlass_utils.sympy","source_any":null,"type_of_any":3}}},"sys":{".class":"SymbolTableNode","cross_ref":"sys","kind":"Gdef"},"time":{".class":"SymbolTableNode","cross_ref":"time","kind":"Gdef"},"torch":{".class":"SymbolTableNode","cross_ref":"torch","kind":"Gdef"},"torch_dtype_to_cutlass_type":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["torch_dtype"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.torch_dtype_to_cutlass_type","name":"torch_dtype_to_cutlass_type","original_first_arg":"torch_dtype","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["torch_dtype"],"arg_types":["torch._C.dtype"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"torch_dtype_to_cutlass_type","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.torch_dtype_to_cutlass_type","name":"torch_dtype_to_cutlass_type","setter_type":null,"type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6},"type_of_any":7}],"extra_attrs":null,"type_ref":"functools._lru_cache_wrapper"}}}},"try_import_cutlass":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[],"arg_names":[],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.try_import_cutlass","name":"try_import_cutlass","original_first_arg":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"try_import_cutlass","ret_type":"builtins.bool","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.cutlass_utils.try_import_cutlass","name":"try_import_cutlass","setter_type":null,"type":{".class":"Instance","args":["builtins.bool"],"extra_attrs":null,"type_ref":"functools._lru_cache_wrapper"}}}}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_utils.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/cutlass_utils.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959414,"dep_lines":[23,20,22,16,18,19,21,470,18,468,2,3,4,5,6,7,8,9,10,11,15,468,1,1,1,1,13],"dep_prios":[5,5,5,5,10,5,5,20,20,20,10,10,10,10,10,10,10,5,5,5,10,20,5,30,30,30,10],"dependencies":["torch._inductor.codegen.cuda.cuda_env","torch._inductor.runtime.runtime_utils","torch._inductor.codegen.cpp_utils","torch._inductor.utils","torch._inductor.config","torch._inductor.ir","torch._inductor.virtualized","torch._inductor.codecache","torch._inductor","unittest.mock","atexit","functools","logging","os","shutil","sys","time","dataclasses","pathlib","typing","torch","unittest","builtins","_frozen_importlib","abc","torch._C"],"hash":"c7bf5ddc71ccc39dfba2045eb56314860856dd8b","id":"torch._inductor.codegen.cuda.cutlass_utils","ignore_all":true,"interface_hash":"a53ebd08ab571247006cc0706b1577fa25d90ff0","mtime":1757363945,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\cutlass_utils.py","plugin_data":null,"size":17539,"suppressed":["sympy"],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/gemm_template.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959427,"dep_lines":[1404,1499,13,32,33,34,35,36,1499,31,32,12,14,15,16,18,19,30,12,18,2,3,4,5,6,7,8,9,11,1,1,1,1,626,937,626],"dep_prios":[20,20,5,5,5,5,5,5,20,5,20,10,5,5,5,5,5,5,20,20,10,10,10,10,10,10,5,5,10,5,30,30,30,20,20,20],"dependencies":["torch._inductor.codegen.cuda.cutlass_lib_extensions.evt_extensions","torch._inductor.codegen.cuda.cutlass_lib_extensions.gemm_operation_extensions","torch._inductor.codegen.cuda.cutlass_cache","torch._inductor.codegen.cuda.cutlass_utils","torch._inductor.codegen.cuda.cuda_kernel","torch._inductor.codegen.cuda.cuda_template","torch._inductor.codegen.cuda.cutlass_presets","torch._inductor.codegen.cuda.cutlass_python_evt","torch._inductor.codegen.cuda.cutlass_lib_extensions","torch._inductor.codegen.common","torch._inductor.codegen.cuda","torch.utils._pytree","torch._inductor.scheduler","torch._inductor.select_algorithm","torch._inductor.utils","torch._inductor.ir","torch._inductor.config","torch._inductor.virtualized","torch.utils","torch._inductor","copy","enum","functools","logging","re","time","abc","typing","torch","builtins","_collections_abc","_frozen_importlib","torch._C"],"hash":"ff4fa0a3af33f2ad93e81d286f3f06a5fdea9e5a","id":"torch._inductor.codegen.cuda.gemm_template","ignore_all":true,"interface_hash":"3c28dcf284dbbb658509fdc9672d51720b2f4d0f","mtime":1757363945,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\gemm_template.py","plugin_data":null,"size":76765,"suppressed":["cutlass_library.library","cutlass_library.gemm_operation","cutlass_library"],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/serialization.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.cuda.serialization","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","CUTLASSOperationSerializer":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer","name":"CUTLASSOperationSerializer","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.cuda.serialization","mro":["torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer","builtins.object"],"names":{".class":"SymbolTable","_SUPPORTED_CLASSES":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","has_explicit_value"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._SUPPORTED_CLASSES","name":"_SUPPORTED_CLASSES","setter_type":null,"type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"}}},"_enum_to_json":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["cls","enum_value"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._enum_to_json","name":"_enum_to_json","original_first_arg":"cls","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_classmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._enum_to_json","name":"_enum_to_json","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","enum_value"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_enum_to_json of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"_gemm_operation_to_json":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["cls","operation"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._gemm_operation_to_json","name":"_gemm_operation_to_json","original_first_arg":"cls","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_classmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._gemm_operation_to_json","name":"_gemm_operation_to_json","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","operation"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_gemm_operation_to_json of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"_json_to_enum":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["cls","json_dict","enum_class"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._json_to_enum","name":"_json_to_enum","original_first_arg":"cls","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_classmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._json_to_enum","name":"_json_to_enum","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["cls","json_dict","enum_class"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_json_to_enum of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"_json_to_gemm_operation":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["cls","json_dict"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._json_to_gemm_operation","name":"_json_to_gemm_operation","original_first_arg":"cls","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_classmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._json_to_gemm_operation","name":"_json_to_gemm_operation","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","json_dict"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_json_to_gemm_operation of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"_json_to_tensor_description":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["cls","tensor_json"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._json_to_tensor_description","name":"_json_to_tensor_description","original_first_arg":"cls","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_classmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._json_to_tensor_description","name":"_json_to_tensor_description","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","tensor_json"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_json_to_tensor_description of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"_json_to_tile_description":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["cls","json_dict"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._json_to_tile_description","name":"_json_to_tile_description","original_first_arg":"cls","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_classmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._json_to_tile_description","name":"_json_to_tile_description","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","json_dict"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_json_to_tile_description of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"_tensor_description_to_json":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["cls","tensor_desc"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._tensor_description_to_json","name":"_tensor_description_to_json","original_first_arg":"cls","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_classmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._tensor_description_to_json","name":"_tensor_description_to_json","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","tensor_desc"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_tensor_description_to_json of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"_tile_description_to_json":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["cls","tile_desc"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._tile_description_to_json","name":"_tile_description_to_json","original_first_arg":"cls","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_classmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer._tile_description_to_json","name":"_tile_description_to_json","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","tile_desc"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_tile_description_to_json of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"deserialize":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["cls","json_str"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated","is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer.deserialize","name":"deserialize","original_first_arg":"cls","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","json_str"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},"builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"deserialize of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_classmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer.deserialize","name":"deserialize","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","json_str"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},"builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"deserialize of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"serialize":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["cls","operation"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated","is_trivial_self"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer.serialize","name":"serialize","original_first_arg":"cls","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","operation"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"serialize of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_classmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer.serialize","name":"serialize","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["cls","operation"],"arg_types":[{".class":"TypeType","item":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"serialize of CUTLASSOperationSerializer","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"Enum":{".class":"SymbolTableNode","cross_ref":"enum.Enum","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.serialization.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.serialization.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.serialization.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.serialization.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.serialization.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.cuda.serialization.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"enum":{".class":"SymbolTableNode","cross_ref":"enum","kind":"Gdef"},"functools":{".class":"SymbolTableNode","cross_ref":"functools","kind":"Gdef"},"get_cutlass_operation_serializer":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[],"arg_names":[],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.cuda.serialization.get_cutlass_operation_serializer","name":"get_cutlass_operation_serializer","original_first_arg":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_cutlass_operation_serializer","ret_type":{".class":"UnionType","items":["torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer",{".class":"NoneType"}],"uses_pep604_syntax":false},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.cuda.serialization.get_cutlass_operation_serializer","name":"get_cutlass_operation_serializer","setter_type":null,"type":{".class":"Instance","args":[{".class":"UnionType","items":["torch._inductor.codegen.cuda.serialization.CUTLASSOperationSerializer",{".class":"NoneType"}],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"functools._lru_cache_wrapper"}}}},"json":{".class":"SymbolTableNode","cross_ref":"json","kind":"Gdef"},"try_import_cutlass":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cuda.cutlass_utils.try_import_cutlass","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\serialization.py"}

.mypy_cache/3.13/torch/_inductor/codegen/cuda/serialization.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959414,"dep_lines":[8,2,3,4,6,1,1,1,71,134,133],"dep_prios":[5,5,10,10,5,5,30,30,20,20,20],"dependencies":["torch._inductor.codegen.cuda.cutlass_utils","enum","functools","json","typing","builtins","_frozen_importlib","abc"],"hash":"733977a66f549532e47de439a8ad51ce04a6347c","id":"torch._inductor.codegen.cuda.serialization","ignore_all":true,"interface_hash":"af3eebd49a7cf40560b592b70bb7bca394a8c8d8","mtime":1757363945,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\cuda\\serialization.py","plugin_data":null,"size":16934,"suppressed":["cutlass_library.library","cutlass_library.gemm_operation","cutlass_library"],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/__init__.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.rocm","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__path__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.__path__","name":"__path__","setter_type":null,"type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"}}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\__init__.py"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/__init__.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757957970,"dep_lines":[1,1,1,1],"dep_prios":[5,30,30,30],"dependencies":["builtins","_frozen_importlib","abc","typing"],"hash":"da39a3ee5e6b4b0d3255bfef95601890afd80709","id":"torch._inductor.codegen.rocm","ignore_all":true,"interface_hash":"485d7444eaecf866dcfa2fdd84f8f747857989e5","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\__init__.py","plugin_data":null,"size":0,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/ck_template.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.rocm.ck_template","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"ArgInfo":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_template.ArgInfo","kind":"Gdef"},"CKTemplate":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.rocm.rocm_template.ROCmTemplate"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate","name":"CKTemplate","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.rocm.ck_template","mro":["torch._inductor.codegen.rocm.ck_template.CKTemplate","torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","torch._inductor.codegen.common.KernelTemplate","builtins.object"],"names":{".class":"SymbolTable","_TORCH_DTYPE_TO_CK":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate._TORCH_DTYPE_TO_CK","name":"_TORCH_DTYPE_TO_CK","setter_type":null,"type":{".class":"Instance","args":["torch._C.dtype","builtins.str"],"extra_attrs":null,"type_ref":"builtins.dict"}}},"get_runtime_arg_info":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated","is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate.get_runtime_arg_info","name":"get_runtime_arg_info","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_template.CKTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_info of CKTemplate","ret_type":{".class":"Instance","args":["torch._inductor.codegen.rocm.rocm_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate.get_runtime_arg_info","name":"get_runtime_arg_info","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_template.CKTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_info of CKTemplate","ret_type":{".class":"Instance","args":["torch._inductor.codegen.rocm.rocm_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"get_runtime_arg_values":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,4],"arg_names":["self","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated","is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate.get_runtime_arg_values","name":"get_runtime_arg_values","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,4],"arg_names":["self","kwargs"],"arg_types":["torch._inductor.codegen.rocm.ck_template.CKTemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_values of CKTemplate","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate.get_runtime_arg_values","name":"get_runtime_arg_values","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,4],"arg_names":["self","kwargs"],"arg_types":["torch._inductor.codegen.rocm.ck_template.CKTemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_values of CKTemplate","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"globals":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate.globals","name":"globals","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_template.CKTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"globals of CKTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"header":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate.header","name":"header","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_template.CKTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"header of CKTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"torch_type_to_ck":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","node","ptr"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate.torch_type_to_ck","name":"torch_type_to_ck","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","node","ptr"],"arg_types":["torch._inductor.codegen.rocm.ck_template.CKTemplate","torch._inductor.ir.IRNode","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"torch_type_to_ck of CKTemplate","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_template.CKTemplate.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.ck_template.CKTemplate","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"IRNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.IRNode","kind":"Gdef"},"IndentedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.IndentedBuffer","kind":"Gdef"},"ROCmTemplate":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_template.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_template.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_template.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_template.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_template.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_template.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"override":{".class":"SymbolTableNode","cross_ref":"typing.override","kind":"Gdef"},"torch":{".class":"SymbolTableNode","cross_ref":"torch","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\ck_template.py"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/ck_template.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959420,"dep_lines":[5,6,7,1,2,4,1,1,1,1,1],"dep_prios":[5,5,5,5,5,10,5,30,30,30,30],"dependencies":["torch._inductor.codegen.rocm.rocm_template","torch._inductor.ir","torch._inductor.utils","typing","typing_extensions","torch","builtins","_frozen_importlib","abc","torch._C","torch._inductor.codegen.common"],"hash":"e85c5aef2cb135d9aac01150e4aa5f08a3912083","id":"torch._inductor.codegen.rocm.ck_template","ignore_all":true,"interface_hash":"e8c38ac27134f8c83373a6ebe4337a063a931a13","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\ck_template.py","plugin_data":null,"size":3698,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/ck_tile_template.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.rocm.ck_tile_template","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","CKTileTemplate":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.rocm.rocm_template.ROCmTemplate"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate","name":"CKTileTemplate","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.rocm.ck_tile_template","mro":["torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate","torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","torch._inductor.codegen.common.KernelTemplate","builtins.object"],"names":{".class":"SymbolTable","_TORCH_DTYPE_TO_CK":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate._TORCH_DTYPE_TO_CK","name":"_TORCH_DTYPE_TO_CK","setter_type":null,"type":{".class":"Instance","args":["torch._C.dtype","builtins.str"],"extra_attrs":null,"type_ref":"builtins.dict"}}},"ck_dtype_to_size":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate.ck_dtype_to_size","name":"ck_dtype_to_size","setter_type":null,"type":{".class":"Instance","args":["builtins.str","builtins.int"],"extra_attrs":null,"type_ref":"builtins.dict"}}},"globals":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate.globals","name":"globals","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"globals of CKTileTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"header":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate.header","name":"header","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"header of CKTileTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"torch_type_to_ck":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","node","ptr"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate.torch_type_to_ck","name":"torch_type_to_ck","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","node","ptr"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate","torch._inductor.ir.IRNode","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"torch_type_to_ck of CKTileTemplate","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"IRNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.IRNode","kind":"Gdef"},"IndentedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.IndentedBuffer","kind":"Gdef"},"ROCmTemplate":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_template.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"torch":{".class":"SymbolTableNode","cross_ref":"torch","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\ck_tile_template.py"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/ck_tile_template.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959420,"dep_lines":[2,3,4,1,1,1,1,1,1,1],"dep_prios":[5,5,5,10,5,30,30,30,30,30],"dependencies":["torch._inductor.codegen.rocm.rocm_template","torch._inductor.ir","torch._inductor.utils","torch","builtins","_frozen_importlib","abc","torch._C","torch._inductor.codegen.common","typing"],"hash":"e1bfc30e7038091a6fec728edf8dda43c0fe5b8f","id":"torch._inductor.codegen.rocm.ck_tile_template","ignore_all":true,"interface_hash":"4099b2d586c65e6bdfadf46485a7cb3451f334dd","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\ck_tile_template.py","plugin_data":null,"size":1542,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/ck_tile_universal_gemm_template.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"ArgInfo":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_template.ArgInfo","kind":"Gdef"},"Buffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Buffer","kind":"Gdef"},"CKTileGemmOperation":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation","name":"CKTileGemmOperation","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation","has_param_spec_type":false,"metaclass_type":null,"metadata":{"dataclass":{"attributes":[{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":39,"name":"layout_a","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":40,"name":"layout_b","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":41,"name":"layout_c","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":43,"name":"datatype_a","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":44,"name":"datatype_b","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":45,"name":"datatype_c","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":47,"name":"tile_m","type":"builtins.int"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":48,"name":"tile_n","type":"builtins.int"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":49,"name":"tile_k","type":"builtins.int"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":51,"name":"warp_m","type":"builtins.int"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":52,"name":"warp_n","type":"builtins.int"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":53,"name":"warp_k","type":"builtins.int"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":55,"name":"warp_tile_m","type":"builtins.int"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":56,"name":"warp_tile_n","type":"builtins.int"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":57,"name":"warp_tile_k","type":"builtins.int"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":59,"name":"m_is_padded","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":60,"name":"n_is_padded","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":61,"name":"k_is_padded","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":63,"name":"pipeline","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":64,"name":"scheduler","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":65,"name":"epilogue","type":"builtins.str"}],"frozen":false},"dataclass_tag":{}},"module_name":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template","mro":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation","builtins.object"],"names":{".class":"SymbolTable","__dataclass_fields__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_classvar","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.__dataclass_fields__","name":"__dataclass_fields__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"dataclasses.Field"}],"extra_attrs":null,"type_ref":"builtins.dict"}},"plugin_generated":true},"__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],"arg_names":["self","layout_a","layout_b","layout_c","datatype_a","datatype_b","datatype_c","tile_m","tile_n","tile_k","warp_m","warp_n","warp_k","warp_tile_m","warp_tile_n","warp_tile_k","m_is_padded","n_is_padded","k_is_padded","pipeline","scheduler","epilogue"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],"arg_names":["self","layout_a","layout_b","layout_c","datatype_a","datatype_b","datatype_c","tile_m","tile_n","tile_k","warp_m","warp_n","warp_k","warp_tile_m","warp_tile_n","warp_tile_k","m_is_padded","n_is_padded","k_is_padded","pipeline","scheduler","epilogue"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of CKTileGemmOperation","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"__match_args__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_ready","allow_incompatible_override"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.__match_args__","name":"__match_args__","setter_type":null,"type":{".class":"TupleType","implicit":false,"items":[{".class":"LiteralType","fallback":"builtins.str","value":"layout_a"},{".class":"LiteralType","fallback":"builtins.str","value":"layout_b"},{".class":"LiteralType","fallback":"builtins.str","value":"layout_c"},{".class":"LiteralType","fallback":"builtins.str","value":"datatype_a"},{".class":"LiteralType","fallback":"builtins.str","value":"datatype_b"},{".class":"LiteralType","fallback":"builtins.str","value":"datatype_c"},{".class":"LiteralType","fallback":"builtins.str","value":"tile_m"},{".class":"LiteralType","fallback":"builtins.str","value":"tile_n"},{".class":"LiteralType","fallback":"builtins.str","value":"tile_k"},{".class":"LiteralType","fallback":"builtins.str","value":"warp_m"},{".class":"LiteralType","fallback":"builtins.str","value":"warp_n"},{".class":"LiteralType","fallback":"builtins.str","value":"warp_k"},{".class":"LiteralType","fallback":"builtins.str","value":"warp_tile_m"},{".class":"LiteralType","fallback":"builtins.str","value":"warp_tile_n"},{".class":"LiteralType","fallback":"builtins.str","value":"warp_tile_k"},{".class":"LiteralType","fallback":"builtins.str","value":"m_is_padded"},{".class":"LiteralType","fallback":"builtins.str","value":"n_is_padded"},{".class":"LiteralType","fallback":"builtins.str","value":"k_is_padded"},{".class":"LiteralType","fallback":"builtins.str","value":"pipeline"},{".class":"LiteralType","fallback":"builtins.str","value":"scheduler"},{".class":"LiteralType","fallback":"builtins.str","value":"epilogue"}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}}},"plugin_generated":true},"__mypy-replace":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5],"arg_names":["layout_a","layout_b","layout_c","datatype_a","datatype_b","datatype_c","tile_m","tile_n","tile_k","warp_m","warp_n","warp_k","warp_tile_m","warp_tile_n","warp_tile_k","m_is_padded","n_is_padded","k_is_padded","pipeline","scheduler","epilogue"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.__mypy-replace","name":"__mypy-replace","original_first_arg":"layout_a","type":{".class":"CallableType","arg_kinds":[5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5],"arg_names":["layout_a","layout_b","layout_c","datatype_a","datatype_b","datatype_c","tile_m","tile_n","tile_k","warp_m","warp_n","warp_k","warp_tile_m","warp_tile_n","warp_tile_k","m_is_padded","n_is_padded","k_is_padded","pipeline","scheduler","epilogue"],"arg_types":["builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-replace of CKTileGemmOperation","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_staticmethod","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.__mypy-replace","name":"__mypy-replace","setter_type":null,"type":{".class":"CallableType","arg_kinds":[5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5],"arg_names":["layout_a","layout_b","layout_c","datatype_a","datatype_b","datatype_c","tile_m","tile_n","tile_k","warp_m","warp_n","warp_k","warp_tile_m","warp_tile_n","warp_tile_k","m_is_padded","n_is_padded","k_is_padded","pipeline","scheduler","epilogue"],"arg_types":["builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-replace of CKTileGemmOperation","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"plugin_generated":true},"__replace__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5],"arg_names":["self","layout_a","layout_b","layout_c","datatype_a","datatype_b","datatype_c","tile_m","tile_n","tile_k","warp_m","warp_n","warp_k","warp_tile_m","warp_tile_n","warp_tile_k","m_is_padded","n_is_padded","k_is_padded","pipeline","scheduler","epilogue"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.__replace__","name":"__replace__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5],"arg_names":["self","layout_a","layout_b","layout_c","datatype_a","datatype_b","datatype_c","tile_m","tile_n","tile_k","warp_m","warp_n","warp_k","warp_tile_m","warp_tile_n","warp_tile_k","m_is_padded","n_is_padded","k_is_padded","pipeline","scheduler","epilogue"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.int","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__replace__ of CKTileGemmOperation","ret_type":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"datatype_a":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.datatype_a","name":"datatype_a","setter_type":null,"type":"builtins.str"}},"datatype_b":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.datatype_b","name":"datatype_b","setter_type":null,"type":"builtins.str"}},"datatype_c":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.datatype_c","name":"datatype_c","setter_type":null,"type":"builtins.str"}},"dict_items":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.dict_items","name":"dict_items","original_first_arg":"self","type":null}},"dtype_repr":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.dtype_repr","name":"dtype_repr","original_first_arg":"self","type":null}},"epilogue":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.epilogue","name":"epilogue","setter_type":null,"type":"builtins.str"}},"k_is_padded":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.k_is_padded","name":"k_is_padded","setter_type":null,"type":"builtins.str"}},"layout_a":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.layout_a","name":"layout_a","setter_type":null,"type":"builtins.str"}},"layout_b":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.layout_b","name":"layout_b","setter_type":null,"type":"builtins.str"}},"layout_c":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.layout_c","name":"layout_c","setter_type":null,"type":"builtins.str"}},"layout_repr":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.layout_repr","name":"layout_repr","original_first_arg":"self","type":null}},"m_is_padded":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.m_is_padded","name":"m_is_padded","setter_type":null,"type":"builtins.str"}},"n_is_padded":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.n_is_padded","name":"n_is_padded","setter_type":null,"type":"builtins.str"}},"name":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.name","name":"name","original_first_arg":"self","type":null}},"pipeline":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.pipeline","name":"pipeline","setter_type":null,"type":"builtins.str"}},"scheduler":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.scheduler","name":"scheduler","setter_type":null,"type":"builtins.str"}},"tile_k":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.tile_k","name":"tile_k","setter_type":null,"type":"builtins.int"}},"tile_m":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.tile_m","name":"tile_m","setter_type":null,"type":"builtins.int"}},"tile_n":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.tile_n","name":"tile_n","setter_type":null,"type":"builtins.int"}},"tile_sizes":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.tile_sizes","name":"tile_sizes","original_first_arg":"self","type":null}},"warp_k":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.warp_k","name":"warp_k","setter_type":null,"type":"builtins.int"}},"warp_m":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.warp_m","name":"warp_m","setter_type":null,"type":"builtins.int"}},"warp_n":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.warp_n","name":"warp_n","setter_type":null,"type":"builtins.int"}},"warp_tile_k":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.warp_tile_k","name":"warp_tile_k","setter_type":null,"type":"builtins.int"}},"warp_tile_m":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.warp_tile_m","name":"warp_tile_m","setter_type":null,"type":"builtins.int"}},"warp_tile_n":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.warp_tile_n","name":"warp_tile_n","setter_type":null,"type":"builtins.int"}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"CKTileGemmTemplate":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","name":"CKTileGemmTemplate","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template","mro":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate","torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","torch._inductor.codegen.common.KernelTemplate","builtins.object"],"names":{".class":"SymbolTable","__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","input_nodes","layout"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","input_nodes","layout"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate",{".class":"Instance","args":["torch._inductor.ir.Buffer"],"extra_attrs":null,"type_ref":"builtins.list"},"torch._inductor.ir.Layout"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of CKTileGemmTemplate","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"add_choices":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["choices","layout","input_nodes"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.add_choices","name":"add_choices","original_first_arg":"choices","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.add_choices","name":"add_choices","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["choices","layout","input_nodes"],"arg_types":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"add_choices of CKTileGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"check_alignments":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.check_alignments","name":"check_alignments","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"check_alignments of CKTileGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"check_block_tile_size":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.check_block_tile_size","name":"check_block_tile_size","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"check_block_tile_size of CKTileGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"check_block_tiles":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.check_block_tiles","name":"check_block_tiles","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"check_block_tiles of CKTileGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"check_dtypes":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.check_dtypes","name":"check_dtypes","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"check_dtypes of CKTileGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"check_layouts":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.check_layouts","name":"check_layouts","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"check_layouts of CKTileGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"check_warp_tiles":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.check_warp_tiles","name":"check_warp_tiles","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"check_warp_tiles of CKTileGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"emit_ck_instance":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.emit_ck_instance","name":"emit_ck_instance","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"emit_ck_instance of CKTileGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"filter_op":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.filter_op","name":"filter_op","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"filter_op of CKTileGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"gemm_template":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.gemm_template","name":"gemm_template","setter_type":null,"type":"builtins.str"}},"gen_ops":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.gen_ops","name":"gen_ops","original_first_arg":"self","type":null}},"get_gemm_problem_size":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.get_gemm_problem_size","name":"get_gemm_problem_size","original_first_arg":"self","type":null}},"get_runtime_arg_info":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.get_runtime_arg_info","name":"get_runtime_arg_info","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_info of CKTileGemmTemplate","ret_type":{".class":"Instance","args":["torch._inductor.codegen.rocm.rocm_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_runtime_arg_values":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,4],"arg_names":["self","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.get_runtime_arg_values","name":"get_runtime_arg_values","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,4],"arg_names":["self","kwargs"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_values of CKTileGemmTemplate","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"globals":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.globals","name":"globals","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"globals of CKTileGemmTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"header":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.header","name":"header","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"header of CKTileGemmTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"k_batch_choices":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.k_batch_choices","name":"k_batch_choices","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"k_batch_choices of CKTileGemmTemplate","ret_type":{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"builtins.tuple"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"render":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,4],"arg_names":["self","kernel","op","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.render","name":"render","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,4],"arg_names":["self","kernel","op","kwargs"],"arg_types":["torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel","torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmOperation",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"render of CKTileGemmTemplate","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"size_args":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.size_args","name":"size_args","original_first_arg":"self","type":null}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.CKTileGemmTemplate","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"CKTileTemplate":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.ck_tile_template.CKTileTemplate","kind":"Gdef"},"IndentedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.IndentedBuffer","kind":"Gdef"},"Layout":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Layout","kind":"Gdef"},"OrderedSet":{".class":"SymbolTableNode","cross_ref":"torch.utils._ordered_set.OrderedSet","kind":"Gdef"},"ROCmTemplateKernel":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"asdict":{".class":"SymbolTableNode","cross_ref":"dataclasses.asdict","kind":"Gdef"},"config":{".class":"SymbolTableNode","cross_ref":"torch._inductor.config","kind":"Gdef"},"dataclass":{".class":"SymbolTableNode","cross_ref":"dataclasses.dataclass","kind":"Gdef"},"functools":{".class":"SymbolTableNode","cross_ref":"functools","kind":"Gdef"},"is_static_int":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["number"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.is_static_int","name":"is_static_int","original_first_arg":"number","type":null}},"log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.log","name":"log","setter_type":null,"type":"logging.Logger"}},"logging":{".class":"SymbolTableNode","cross_ref":"logging","kind":"Gdef"},"ops":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[],"arg_names":[],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_decorated"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.ops","name":"ops","original_first_arg":null,"type":null},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.ops","name":"ops","setter_type":null,"type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_of_any":7}],"extra_attrs":null,"type_ref":"functools._lru_cache_wrapper"}}}},"random":{".class":"SymbolTableNode","cross_ref":"random","kind":"Gdef"},"torch":{".class":"SymbolTableNode","cross_ref":"torch","kind":"Gdef"},"torch_layout_to_ck_layout":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["torch_layout"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template.torch_layout_to_ck_layout","name":"torch_layout_to_ck_layout","original_first_arg":"torch_layout","type":null}}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\ck_tile_universal_gemm_template.py"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/ck_tile_universal_gemm_template.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959425,"dep_lines":[10,11,12,9,13,14,16,9,2,3,4,5,6,8,1,1,1,1],"dep_prios":[5,5,5,10,5,5,5,20,10,10,10,5,5,10,5,30,30,30],"dependencies":["torch._inductor.codegen.rocm.ck_tile_template","torch._inductor.codegen.rocm.rocm_kernel","torch._inductor.codegen.rocm.rocm_template","torch._inductor.config","torch._inductor.ir","torch.utils._ordered_set","torch._inductor.utils","torch._inductor","functools","logging","random","dataclasses","typing","torch","builtins","_frozen_importlib","abc","torch._inductor.codegen.common"],"hash":"60fb8d4f6c83c6ab0047f458c41230a621fe9c98","id":"torch._inductor.codegen.rocm.ck_tile_universal_gemm_template","ignore_all":true,"interface_hash":"df5de07848ba732caaccddf9d3d76c1b1671c72d","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\ck_tile_universal_gemm_template.py","plugin_data":null,"size":37323,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/ck_universal_gemm_template.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Buffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Buffer","kind":"Gdef"},"CKGemmOperation":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmOperation","name":"CKGemmOperation","setter_type":null,"type":{".class":"TypeType","item":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}}}},"CKGemmTemplate":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.rocm.ck_template.CKTemplate"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate","name":"CKGemmTemplate","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.rocm.ck_universal_gemm_template","mro":["torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate","torch._inductor.codegen.rocm.ck_template.CKTemplate","torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","torch._inductor.codegen.common.KernelTemplate","builtins.object"],"names":{".class":"SymbolTable","__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,0,1],"arg_names":["self","input_nodes","layout","alpha","beta","input_reorder"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,0,1],"arg_names":["self","input_nodes","layout","alpha","beta","input_reorder"],"arg_types":["torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate",{".class":"Instance","args":["torch._inductor.ir.Buffer"],"extra_attrs":null,"type_ref":"builtins.list"},"torch._inductor.ir.Layout","builtins.float","builtins.float",{".class":"UnionType","items":[{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of CKGemmTemplate","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_check_num_k_loops":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","op","kBatch"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate._check_num_k_loops","name":"_check_num_k_loops","original_first_arg":"self","type":null}},"_get_kBatch":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate._get_kBatch","name":"_get_kBatch","original_first_arg":"self","type":null}},"_has_padding":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","dimension","gemm_specialization"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate._has_padding","name":"_has_padding","original_first_arg":"self","type":null}},"_is_rcr_f16":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate._is_rcr_f16","name":"_is_rcr_f16","original_first_arg":"self","type":null}},"_prefetch_stages":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,1],"arg_names":["self","op","a_dtype_size","b_dtype_size","warp_size"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate._prefetch_stages","name":"_prefetch_stages","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,1],"arg_names":["self","op","a_dtype_size","b_dtype_size","warp_size"],"arg_types":["torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_prefetch_stages of CKGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"add_ck_gemm_choices":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,1,1,1],"arg_names":["choices","layout","input_nodes","alpha","beta","input_reorder"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.add_ck_gemm_choices","name":"add_ck_gemm_choices","original_first_arg":"choices","type":null},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.add_ck_gemm_choices","name":"add_ck_gemm_choices","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0,0,1,1,1],"arg_names":["choices","layout","input_nodes","alpha","beta","input_reorder"],"arg_types":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":true,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"add_ck_gemm_choices of CKGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"alpha":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.alpha","name":"alpha","setter_type":null,"type":"builtins.float"}},"beta":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.beta","name":"beta","setter_type":null,"type":"builtins.float"}},"emit_ck_instance":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.emit_ck_instance","name":"emit_ck_instance","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op"],"arg_types":["torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate",{".class":"UnboundType","args":[],"expr":"CKGemmOperation","expr_fallback":"builtins.str","name":"CKGemmOperation"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"emit_ck_instance of CKGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"filter_op":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","op_info"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.filter_op","name":"filter_op","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["self","op_info"],"arg_types":["torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate",{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"filter_op of CKGemmTemplate","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"gemm_template":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.gemm_template","name":"gemm_template","setter_type":null,"type":"builtins.str"}},"gen_ops":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.gen_ops","name":"gen_ops","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"gen_ops of CKGemmTemplate","ret_type":{".class":"Instance","args":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp"}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"globals":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.globals","name":"globals","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"globals of CKGemmTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"header":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.header","name":"header","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"header of CKGemmTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"inline_utils":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.inline_utils","name":"inline_utils","original_first_arg":"self","type":null}},"is_batched":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.is_batched","name":"is_batched","setter_type":null,"type":"builtins.bool"}},"render":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,4],"arg_names":["self","kernel","op","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.render","name":"render","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,4],"arg_names":["self","kernel","op","kwargs"],"arg_types":["torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate","torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel",{".class":"UnboundType","args":[],"expr":"CKGemmOperation","expr_fallback":"builtins.str","name":"CKGemmOperation"},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"render of CKGemmTemplate","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"size_args":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.size_args","name":"size_args","original_first_arg":"self","type":null}},"standalone_runner_template":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.standalone_runner_template","name":"standalone_runner_template","setter_type":null,"type":"builtins.str"}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.ck_universal_gemm_template.CKGemmTemplate","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"CKTemplate":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.ck_template.CKTemplate","kind":"Gdef"},"DTYPE_TO_CPP":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cpp_utils.DTYPE_TO_CPP","kind":"Gdef"},"IndentedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.IndentedBuffer","kind":"Gdef"},"InductorROCmOp":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":[{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp","name":"InductorROCmOp","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":["is_named_tuple"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp","has_param_spec_type":false,"metaclass_type":null,"metadata":{"namedtuple":{"fields":["op","kBatch"]}},"module_name":"torch._inductor.codegen.rocm.ck_universal_gemm_template","mro":["torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp","builtins.tuple","typing.Sequence","typing.Reversible","typing.Collection","typing.Iterable","typing.Container","builtins.object"],"names":{".class":"SymbolTable","_NT":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"TypeVarExpr","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","name":"_NT","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}},"__annotations__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__match_args__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__match_args__","name":"__match_args__","setter_type":null,"type":{".class":"TupleType","implicit":false,"items":[{".class":"LiteralType","fallback":"builtins.str","value":"op"},{".class":"LiteralType","fallback":"builtins.str","value":"kBatch"}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}}}},"__new__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["_cls","op","kBatch"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__new__","name":"__new__","original_first_arg":"_cls","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["_cls","op","kBatch"],"arg_types":[{".class":"TypeType","item":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__new__","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__new__ of InductorROCmOp","ret_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__new__","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__new__","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}]}},"plugin_generated":true},"__replace__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,5,5],"arg_names":["_self","op","kBatch"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__replace__","name":"__replace__","original_first_arg":"_self","type":{".class":"CallableType","arg_kinds":[0,5,5],"arg_names":["_self","op","kBatch"],"arg_types":[{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__replace__","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__replace__ of InductorROCmOp","ret_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__replace__","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.__replace__","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}]}},"plugin_generated":true},"_asdict":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["_self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._asdict","name":"_asdict","original_first_arg":"_self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["_self"],"arg_types":[{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._asdict","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_asdict of InductorROCmOp","ret_type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._asdict","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}]}},"plugin_generated":true},"_field_defaults":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._field_defaults","name":"_field_defaults","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"_field_types":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._field_types","name":"_field_types","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"_fields":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._fields","name":"_fields","setter_type":null,"type":{".class":"TupleType","implicit":false,"items":["builtins.str","builtins.str"],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}}}},"_make":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["_cls","iterable"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_class","is_decorated"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._make","name":"_make","original_first_arg":"_cls","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["_cls","iterable"],"arg_types":[{".class":"TypeType","item":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._make","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}},{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"typing.Iterable"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_make of InductorROCmOp","ret_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._make","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._make","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}]}},"is_overload":false,"var":{".class":"Var","flags":["is_classmethod","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._make","name":"_make","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["_cls","iterable"],"arg_types":[{".class":"TypeType","item":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._make","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}},{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"typing.Iterable"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_make of InductorROCmOp","ret_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._make","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._make","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}]}}},"plugin_generated":true},"_replace":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,5,5],"arg_names":["_self","op","kBatch"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._replace","name":"_replace","original_first_arg":"_self","type":{".class":"CallableType","arg_kinds":[0,5,5],"arg_names":["_self","op","kBatch"],"arg_types":[{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._replace","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_replace of InductorROCmOp","ret_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._replace","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._NT","id":-1,"name":"_NT","namespace":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._replace","upper_bound":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"values":[],"variance":0}]}},"plugin_generated":true},"_source":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp._source","name":"_source","setter_type":null,"type":"builtins.str"}},"kBatch":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_property","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.kBatch","name":"kBatch","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}}},"op":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_property","is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.InductorROCmOp.op","name":"op","setter_type":null,"type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}}}},"self_type":null,"slots":null,"tuple_type":{".class":"TupleType","implicit":false,"items":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"extra_attrs":null,"type_ref":"builtins.tuple"}},"type_vars":[],"typeddict_type":null}},"Layout":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Layout","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"ROCmTemplateKernel":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel","kind":"Gdef"},"_":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template._","name":"_","setter_type":null,"type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false}}},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"config":{".class":"SymbolTableNode","cross_ref":"torch._inductor.config","kind":"Gdef"},"copy":{".class":"SymbolTableNode","cross_ref":"copy","kind":"Gdef"},"gen_ops_library":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.gen_ops_library","name":"gen_ops_library","setter_type":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"gen_ops_preselected":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.gen_ops_preselected","name":"gen_ops_preselected","setter_type":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"is_dynamic":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.is_dynamic","kind":"Gdef"},"is_static_int":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["number"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.is_static_int","name":"is_static_int","original_first_arg":"number","type":null}},"log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.log","name":"log","setter_type":null,"type":"logging.Logger"}},"logging":{".class":"SymbolTableNode","cross_ref":"logging","kind":"Gdef"},"math":{".class":"SymbolTableNode","cross_ref":"math","kind":"Gdef"},"namedtuple":{".class":"SymbolTableNode","cross_ref":"collections.namedtuple","kind":"Gdef"},"next_power_of_2":{".class":"SymbolTableNode","cross_ref":"torch._inductor.runtime.runtime_utils.next_power_of_2","kind":"Gdef"},"padding_lookup":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.padding_lookup","name":"padding_lookup","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"Instance","args":["builtins.str","builtins.bool"],"extra_attrs":null,"type_ref":"builtins.dict"}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"random":{".class":"SymbolTableNode","cross_ref":"random","kind":"Gdef"},"rocm_compile_command":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.compile_command.rocm_compile_command","kind":"Gdef"},"sympy":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_suppressed_import","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.sympy","name":"sympy","setter_type":null,"type":{".class":"AnyType","missing_import_name":"torch._inductor.codegen.rocm.ck_universal_gemm_template.sympy","source_any":null,"type_of_any":3}}},"torch":{".class":"SymbolTableNode","cross_ref":"torch","kind":"Gdef"},"torch_layout_to_ck_layout":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["torch_layout"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.ck_universal_gemm_template.torch_layout_to_ck_layout","name":"torch_layout_to_ck_layout","original_first_arg":"torch_layout","type":null}},"try_import_ck_lib":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.try_import_ck_lib","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\ck_universal_gemm_template.py"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/ck_universal_gemm_template.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959420,"dep_lines":[14,15,16,13,18,12,17,20,12,2,3,4,5,6,7,11,1,1,1,1,1,1,915,918,9],"dep_prios":[5,5,5,5,5,10,5,5,20,10,10,10,10,5,5,10,5,30,30,30,30,30,20,20,10],"dependencies":["torch._inductor.codegen.rocm.ck_template","torch._inductor.codegen.rocm.compile_command","torch._inductor.codegen.rocm.rocm_kernel","torch._inductor.codegen.cpp_utils","torch._inductor.runtime.runtime_utils","torch._inductor.config","torch._inductor.ir","torch._inductor.utils","torch._inductor","copy","logging","math","random","collections","typing","torch","builtins","_frozen_importlib","abc","functools","torch._inductor.codegen.common","torch._inductor.codegen.rocm.rocm_template"],"hash":"6232faad7eac8017667dd30e2c6823993025e0b9","id":"torch._inductor.codegen.rocm.ck_universal_gemm_template","ignore_all":true,"interface_hash":"b5323f9064f3a92a157d023aebef59f89cba7513","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\ck_universal_gemm_template.py","plugin_data":null,"size":40526,"suppressed":["ck4inductor.batched_universal_gemm.gen_instances","ck4inductor.universal_gemm.gen_instances","sympy"],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/compile_command.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.rocm.compile_command","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.compile_command.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.compile_command.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.compile_command.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.compile_command.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.compile_command.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.compile_command.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"_rocm_compiler_options":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[],"arg_names":[],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.compile_command._rocm_compiler_options","name":"_rocm_compiler_options","original_first_arg":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_rocm_compiler_options","ret_type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_rocm_include_paths":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["dst_file_ext"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.compile_command._rocm_include_paths","name":"_rocm_include_paths","original_first_arg":"dst_file_ext","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["dst_file_ext"],"arg_types":["builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_rocm_include_paths","ret_type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_rocm_lib_options":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["dst_file_ext"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.compile_command._rocm_lib_options","name":"_rocm_lib_options","original_first_arg":"dst_file_ext","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["dst_file_ext"],"arg_types":["builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_rocm_lib_options","ret_type":{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"config":{".class":"SymbolTableNode","cross_ref":"torch._inductor.config","kind":"Gdef"},"is_linux":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.is_linux","kind":"Gdef"},"log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.compile_command.log","name":"log","setter_type":null,"type":"logging.Logger"}},"logging":{".class":"SymbolTableNode","cross_ref":"logging","kind":"Gdef"},"os":{".class":"SymbolTableNode","cross_ref":"os","kind":"Gdef"},"rocm_compile_command":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,1],"arg_names":["src_files","dst_file","dst_file_ext","extra_args"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.compile_command.rocm_compile_command","name":"rocm_compile_command","original_first_arg":"src_files","type":{".class":"CallableType","arg_kinds":[0,0,0,1],"arg_names":["src_files","dst_file","dst_file_ext","extra_args"],"arg_types":[{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},"builtins.str","builtins.str",{".class":"UnionType","items":[{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"rocm_compile_command","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"rocm_compiler":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[],"arg_names":[],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.compile_command.rocm_compiler","name":"rocm_compiler","original_first_arg":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"rocm_compiler","ret_type":{".class":"UnionType","items":["builtins.str",{".class":"NoneType"}],"uses_pep604_syntax":false},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\compile_command.py"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/compile_command.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959350,"dep_lines":[6,7,6,2,3,4,1,1,1],"dep_prios":[10,5,20,10,10,5,5,30,30],"dependencies":["torch._inductor.config","torch._inductor.utils","torch._inductor","logging","os","typing","builtins","_frozen_importlib","abc"],"hash":"634a06b864d4e9d073f76fb6e7645a4328495863","id":"torch._inductor.codegen.rocm.compile_command","ignore_all":true,"interface_hash":"34c43f65fa2af961461d18f8ba155a8f1c301471","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\compile_command.py","plugin_data":null,"size":4620,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/rocm_benchmark_request.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request","future_import_flags":["annotations"],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"BenchmarkRequest":{".class":"SymbolTableNode","cross_ref":"torch._inductor.autotune_process.BenchmarkRequest","kind":"Gdef"},"Callable":{".class":"SymbolTableNode","cross_ref":"typing.Callable","kind":"Gdef"},"DLLWrapper":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codecache.DLLWrapper","kind":"Gdef"},"GPUDeviceBenchmarkMixin":{".class":"SymbolTableNode","cross_ref":"torch._inductor.autotune_process.GPUDeviceBenchmarkMixin","kind":"Gdef"},"Iterable":{".class":"SymbolTableNode","cross_ref":"typing.Iterable","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"ROCmBenchmarkRequest":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.autotune_process.GPUDeviceBenchmarkMixin","torch._inductor.autotune_process.BenchmarkRequest"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest","name":"ROCmBenchmarkRequest","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.rocm.rocm_benchmark_request","mro":["torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest","torch._inductor.autotune_process.GPUDeviceBenchmarkMixin","torch._inductor.autotune_process.BenchmarkRequest","builtins.object"],"names":{".class":"SymbolTable","DLL":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.DLL","name":"DLL","setter_type":null,"type":{".class":"UnionType","items":["torch._inductor.codecache.DLLWrapper",{".class":"NoneType"}],"uses_pep604_syntax":false}}},"__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,0,0],"arg_names":["self","kernel_name","input_tensor_meta","output_tensor_meta","extra_args","source_code"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,0,0],"arg_names":["self","kernel_name","input_tensor_meta","output_tensor_meta","extra_args","source_code"],"arg_types":["torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest","builtins.str",{".class":"UnionType","items":["torch._inductor.autotune_process.TensorMeta",{".class":"Instance","args":["torch._inductor.autotune_process.TensorMeta"],"extra_attrs":null,"type_ref":"builtins.list"}],"uses_pep604_syntax":false},{".class":"UnionType","items":["torch._inductor.autotune_process.TensorMeta",{".class":"Instance","args":["torch._inductor.autotune_process.TensorMeta"],"extra_attrs":null,"type_ref":"builtins.list"}],"uses_pep604_syntax":false},{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"typing.Iterable"},"builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of ROCmBenchmarkRequest","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"__str__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":[null],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.__str__","name":"__str__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":[null],"arg_types":["torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__str__ of ROCmBenchmarkRequest","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"_workspace_size_updated":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest._workspace_size_updated","name":"_workspace_size_updated","setter_type":null,"type":"builtins.bool"}},"cleanup_run_fn":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.cleanup_run_fn","name":"cleanup_run_fn","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"cleanup_run_fn of ROCmBenchmarkRequest","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"ensure_dll_loaded":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.ensure_dll_loaded","name":"ensure_dll_loaded","original_first_arg":"self","type":null}},"hash_key":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.hash_key","name":"hash_key","setter_type":null,"type":"builtins.str"}},"make_run_fn":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,2,3],"arg_names":["self","input_tensors","out"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.make_run_fn","name":"make_run_fn","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,2,3],"arg_names":["self","input_tensors","out"],"arg_types":["torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest","torch._tensor.Tensor","torch._tensor.Tensor"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"make_run_fn of ROCmBenchmarkRequest","ret_type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"precompile":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.precompile","name":"precompile","original_first_arg":"self","type":null}},"source_code":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.source_code","name":"source_code","setter_type":null,"type":"builtins.str"}},"source_file":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.source_file","name":"source_file","setter_type":null,"type":"builtins.str"}},"update_workspace_size":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.update_workspace_size","name":"update_workspace_size","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"update_workspace_size of ROCmBenchmarkRequest","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"workspace":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.workspace","name":"workspace","setter_type":null,"type":{".class":"UnionType","items":["torch._tensor.Tensor",{".class":"NoneType"}],"uses_pep604_syntax":false}}},"workspace_size":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.workspace_size","name":"workspace_size","setter_type":null,"type":"builtins.int"}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"ROCmCodeCache":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codecache.ROCmCodeCache","kind":"Gdef"},"TYPE_CHECKING":{".class":"SymbolTableNode","cross_ref":"typing.TYPE_CHECKING","kind":"Gdef"},"TensorMeta":{".class":"SymbolTableNode","cross_ref":"torch._inductor.autotune_process.TensorMeta","kind":"Gdef"},"Union":{".class":"SymbolTableNode","cross_ref":"typing.Union","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"annotations":{".class":"SymbolTableNode","cross_ref":"__future__.annotations","kind":"Gdef"},"byref":{".class":"SymbolTableNode","cross_ref":"_ctypes.byref","kind":"Gdef"},"c_int":{".class":"SymbolTableNode","cross_ref":"ctypes.c_int","kind":"Gdef"},"c_size_t":{".class":"SymbolTableNode","cross_ref":"ctypes.c_size_t","kind":"Gdef"},"c_void_p":{".class":"SymbolTableNode","cross_ref":"ctypes.c_void_p","kind":"Gdef"},"config":{".class":"SymbolTableNode","cross_ref":"torch._inductor.config","kind":"Gdef"},"functools":{".class":"SymbolTableNode","cross_ref":"functools","kind":"Gdef"},"log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.rocm_benchmark_request.log","name":"log","setter_type":null,"type":"logging.Logger"}},"logging":{".class":"SymbolTableNode","cross_ref":"logging","kind":"Gdef"},"torch":{".class":"SymbolTableNode","cross_ref":"torch","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\rocm_benchmark_request.py"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/rocm_benchmark_request.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959412,"dep_lines":[10,11,16,10,20,2,4,5,6,7,9,1,1,1,1,1],"dep_prios":[10,5,5,20,25,5,10,10,5,5,10,5,30,30,30,30],"dependencies":["torch._inductor.config","torch._inductor.autotune_process","torch._inductor.codecache","torch._inductor","collections.abc","__future__","functools","logging","ctypes","typing","torch","builtins","_frozen_importlib","abc","torch._C","torch._tensor"],"hash":"5bf567392273d9e88a2a41c64cf1c92b15a6471e","id":"torch._inductor.codegen.rocm.rocm_benchmark_request","ignore_all":true,"interface_hash":"63aec7b7f12e48237dfbefda0f3a53d25f07160a","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\rocm_benchmark_request.py","plugin_data":null,"size":5214,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/rocm_cpp_scheduling.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","BaseSchedulerNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.scheduler.BaseSchedulerNode","kind":"Gdef"},"BaseScheduling":{".class":"SymbolTableNode","cross_ref":"torch._inductor.scheduler.BaseScheduling","kind":"Gdef"},"IndentedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.IndentedBuffer","kind":"Gdef"},"ROCmCPPScheduling":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.scheduler.BaseScheduling"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling","name":"ROCmCPPScheduling","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.rocm.rocm_cpp_scheduling","mro":["torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling","torch._inductor.scheduler.BaseScheduling","builtins.object"],"names":{".class":"SymbolTable","can_fuse_vertical":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","node1","node2"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling.can_fuse_vertical","name":"can_fuse_vertical","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","node1","node2"],"arg_types":["torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling","torch._inductor.scheduler.BaseSchedulerNode","torch._inductor.scheduler.BaseSchedulerNode"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"can_fuse_vertical of ROCmCPPScheduling","ret_type":"builtins.bool","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"codegen_template":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0],"arg_names":["self","template_node","epilogue_nodes","prologue_nodes"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling.codegen_template","name":"codegen_template","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0],"arg_names":["self","template_node","epilogue_nodes","prologue_nodes"],"arg_types":["torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling","torch._inductor.scheduler.BaseSchedulerNode",{".class":"Instance","args":["torch._inductor.scheduler.BaseSchedulerNode"],"extra_attrs":null,"type_ref":"typing.Sequence"},{".class":"Instance","args":["torch._inductor.scheduler.BaseSchedulerNode"],"extra_attrs":null,"type_ref":"typing.Sequence"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"codegen_template of ROCmCPPScheduling","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"define_kernel":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","src_code","node_schedule"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling.define_kernel","name":"define_kernel","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","src_code","node_schedule"],"arg_types":["torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling","builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"define_kernel of ROCmCPPScheduling","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"group_fn":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["self","sizes"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling.group_fn","name":"group_fn","original_first_arg":"self","type":null}},"is_rocm_cpp_template":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["node"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling.is_rocm_cpp_template","name":"is_rocm_cpp_template","original_first_arg":"node","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["node"],"arg_types":["torch._inductor.scheduler.BaseSchedulerNode"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"is_rocm_cpp_template of ROCmCPPScheduling","ret_type":"builtins.bool","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_initialized_in_class","is_staticmethod","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling.is_rocm_cpp_template","name":"is_rocm_cpp_template","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0],"arg_names":["node"],"arg_types":["torch._inductor.scheduler.BaseSchedulerNode"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"is_rocm_cpp_template of ROCmCPPScheduling","ret_type":"builtins.bool","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.ROCmCPPScheduling","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"ROCmTemplateBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_template_buffer.ROCmTemplateBuffer","kind":"Gdef"},"SchedulerNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.scheduler.SchedulerNode","kind":"Gdef"},"Sequence":{".class":"SymbolTableNode","cross_ref":"typing.Sequence","kind":"Gdef"},"V":{".class":"SymbolTableNode","cross_ref":"torch._inductor.virtualized.V","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"cast":{".class":"SymbolTableNode","cross_ref":"typing.cast","kind":"Gdef"},"code_hash":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codecache.code_hash","kind":"Gdef"},"config":{".class":"SymbolTableNode","cross_ref":"torch._inductor.config","kind":"Gdef"},"get_fused_kernel_name":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.get_fused_kernel_name","kind":"Gdef"},"get_kernel_metadata":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.get_kernel_metadata","kind":"Gdef"},"get_path":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codecache.get_path","kind":"Gdef"},"log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.rocm_cpp_scheduling.log","name":"log","setter_type":null,"type":"logging.Logger"}},"logging":{".class":"SymbolTableNode","cross_ref":"logging","kind":"Gdef"},"sympy_product":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.sympy_product","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\rocm_cpp_scheduling.py"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/rocm_cpp_scheduling.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959410,"dep_lines":[12,11,6,7,8,9,10,3,6,2,4,1,1,1],"dep_prios":[5,5,10,5,5,5,5,5,20,10,5,5,30,30],"dependencies":["torch._inductor.codegen.rocm.rocm_template_buffer","torch._inductor.codegen.common","torch._inductor.config","torch._inductor.codecache","torch._inductor.scheduler","torch._inductor.utils","torch._inductor.virtualized","collections.abc","torch._inductor","logging","typing","builtins","_frozen_importlib","abc"],"hash":"469807916dc1de12bb033cb87c6301691114f365","id":"torch._inductor.codegen.rocm.rocm_cpp_scheduling","ignore_all":true,"interface_hash":"c853711980784a729181e83674b8bee2fb39fae7","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\rocm_cpp_scheduling.py","plugin_data":null,"size":3898,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/rocm_kernel.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.rocm.rocm_kernel","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"ArgInfo":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_template.ArgInfo","kind":"Gdef"},"Buffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Buffer","kind":"Gdef"},"Callable":{".class":"SymbolTableNode","cross_ref":"typing.Callable","kind":"Gdef"},"ChoiceCaller":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.ChoiceCaller","kind":"Gdef"},"CppPrinter":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cpp_utils.CppPrinter","kind":"Gdef"},"CppWrapperCpu":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.cpp_wrapper_cpu.CppWrapperCpu","kind":"Gdef"},"DTYPE_TO_ROCM_TYPE":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_utils.DTYPE_TO_ROCM_TYPE","kind":"Gdef"},"IRNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.IRNode","kind":"Gdef"},"Kernel":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.Kernel","kind":"Gdef"},"Layout":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Layout","kind":"Gdef"},"OpOverrides":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.OpOverrides","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"PrimitiveInfoType":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.PrimitiveInfoType","kind":"Gdef"},"ROCmBenchmarkRequest":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest","kind":"Gdef"},"ROCmKernel":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":[{".class":"Instance","args":["torch._inductor.codegen.common.CSEVariable"],"extra_attrs":null,"type_ref":"torch._inductor.codegen.common.Kernel"}],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmKernel","name":"ROCmKernel","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmKernel","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.rocm.rocm_kernel","mro":["torch._inductor.codegen.rocm.rocm_kernel.ROCmKernel","torch._inductor.codegen.common.Kernel","torch._inductor.codegen.common.CodeGen","builtins.object"],"names":{".class":"SymbolTable","overrides":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmKernel.overrides","name":"overrides","setter_type":null,"type":{".class":"CallableType","arg_kinds":[],"arg_names":[],"arg_types":[],"fallback":"builtins.type","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":true,"is_ellipsis_args":false,"name":null,"ret_type":"torch._inductor.codegen.common.OpOverrides","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmKernel.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.rocm_kernel.ROCmKernel","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"ROCmTemplate":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","kind":"Gdef"},"ROCmTemplateBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_template_buffer.ROCmTemplateBuffer","kind":"Gdef"},"ROCmTemplateCaller":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.ir.ChoiceCaller"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller","name":"ROCmTemplateCaller","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.rocm.rocm_kernel","mro":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller","torch._inductor.ir.ChoiceCaller","builtins.object"],"names":{".class":"SymbolTable","__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,0,0,0,0,0],"arg_names":["self","name","category","input_nodes","layout","make_kernel_render","bmreq","template","info_kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,0,0,0,0,0],"arg_names":["self","name","category","input_nodes","layout","make_kernel_render","bmreq","template","info_kwargs"],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller","builtins.str","builtins.str",{".class":"Instance","args":["torch._inductor.ir.Buffer"],"extra_attrs":null,"type_ref":"builtins.list"},"torch._inductor.ir.Layout",{".class":"CallableType","arg_kinds":[0,0],"arg_names":[null,null],"arg_types":["torch._inductor.codegen.rocm.rocm_template_buffer.ROCmTemplateBuffer",{".class":"UnionType","items":[{".class":"Instance","args":["torch._inductor.ir.IRNode"],"extra_attrs":null,"type_ref":"typing.Sequence"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]},"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest","torch._inductor.codegen.rocm.rocm_template.ROCmTemplate",{".class":"UnionType","items":[{".class":"Instance","args":["builtins.str",{".class":"UnionType","items":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"},{".class":"Instance","args":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"}],"extra_attrs":null,"type_ref":"builtins.list"}],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"builtins.dict"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of ROCmTemplateCaller","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"__str__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":[null],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.__str__","name":"__str__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":[null],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__str__ of ROCmTemplateCaller","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"benchmark":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,2,3],"arg_names":["self","args","out"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.benchmark","name":"benchmark","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,2,3],"arg_names":["self","args","out"],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"benchmark of ROCmTemplateCaller","ret_type":"builtins.float","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"bmreq":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.bmreq","name":"bmreq","setter_type":null,"type":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest"}},"call_name":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.call_name","name":"call_name","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"call_name of ROCmTemplateCaller","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"category":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.category","name":"category","setter_type":null,"type":"builtins.str"}},"hash_key":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.hash_key","name":"hash_key","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"hash_key of ROCmTemplateCaller","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"info_dict":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.info_dict","name":"info_dict","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"info_dict of ROCmTemplateCaller","ret_type":{".class":"Instance","args":["builtins.str",{".class":"UnionType","items":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"},{".class":"Instance","args":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"}],"extra_attrs":null,"type_ref":"builtins.list"}],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"builtins.dict"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"info_kwargs":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.info_kwargs","name":"info_kwargs","setter_type":null,"type":{".class":"UnionType","items":[{".class":"Instance","args":["builtins.str",{".class":"UnionType","items":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"},{".class":"Instance","args":[{".class":"TypeAliasType","args":[],"type_ref":"torch._inductor.ir.PrimitiveInfoType"}],"extra_attrs":null,"type_ref":"builtins.list"}],"uses_pep604_syntax":false}],"extra_attrs":null,"type_ref":"builtins.dict"},{".class":"NoneType"}],"uses_pep604_syntax":false}}},"make_kernel_render":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.make_kernel_render","name":"make_kernel_render","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":[null,null],"arg_types":["torch._inductor.codegen.rocm.rocm_template_buffer.ROCmTemplateBuffer",{".class":"UnionType","items":[{".class":"Instance","args":["torch._inductor.ir.IRNode"],"extra_attrs":null,"type_ref":"typing.Sequence"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":null,"ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"output_node":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.output_node","name":"output_node","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"output_node of ROCmTemplateCaller","ret_type":"torch._inductor.ir.TensorBox","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"precompile":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.precompile","name":"precompile","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"precompile of ROCmTemplateCaller","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"template":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.template","name":"template","setter_type":null,"type":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate"}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"ROCmTemplateKernel":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.rocm.rocm_kernel.ROCmKernel"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel","name":"ROCmTemplateKernel","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.rocm.rocm_kernel","mro":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel","torch._inductor.codegen.rocm.rocm_kernel.ROCmKernel","torch._inductor.codegen.common.Kernel","torch._inductor.codegen.common.CodeGen","builtins.object"],"names":{".class":"SymbolTable","_EXTRA_CPP_ARGS":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel._EXTRA_CPP_ARGS","name":"_EXTRA_CPP_ARGS","setter_type":null,"type":"builtins.str"}},"__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0],"arg_names":["self","kernel_name","runtime_arg_info","runtime_arg_values"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0],"arg_names":["self","kernel_name","runtime_arg_info","runtime_arg_values"],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel","builtins.str",{".class":"Instance","args":["torch._inductor.codegen.rocm.rocm_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of ROCmTemplateKernel","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"call_kernel":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","name","node"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel.call_kernel","name":"call_kernel","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","name","node"],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel","builtins.str","torch._inductor.codegen.rocm.rocm_template_buffer.ROCmTemplateBuffer"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"call_kernel of ROCmTemplateKernel","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"def_kernel":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,1,1],"arg_names":["self","inputs","outputs","size_args","names_str","input_reorder"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel.def_kernel","name":"def_kernel","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,1,1],"arg_names":["self","inputs","outputs","size_args","names_str","input_reorder"],"arg_types":["torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel",{".class":"Instance","args":["torch._inductor.ir.IRNode"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":["torch._inductor.ir.IRNode"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"Instance","args":["builtins.str"],"extra_attrs":null,"type_ref":"builtins.list"},"builtins.str",{".class":"UnionType","items":[{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"def_kernel of ROCmTemplateKernel","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_signature":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel.get_signature","name":"get_signature","original_first_arg":"self","type":null}},"named_nodes":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel.named_nodes","name":"named_nodes","setter_type":null,"type":{".class":"Instance","args":["builtins.str","torch._inductor.ir.IRNode"],"extra_attrs":null,"type_ref":"builtins.dict"}}},"runtime_arg_info":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel.runtime_arg_info","name":"runtime_arg_info","setter_type":null,"type":{".class":"Instance","args":["torch._inductor.codegen.rocm.rocm_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"}}},"runtime_arg_values":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel.runtime_arg_values","name":"runtime_arg_values","setter_type":null,"type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"}}},"signature":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel.signature","name":"signature","setter_type":null,"type":"builtins.str"}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"Sequence":{".class":"SymbolTableNode","cross_ref":"typing.Sequence","kind":"Gdef"},"TYPE_CHECKING":{".class":"SymbolTableNode","cross_ref":"typing.TYPE_CHECKING","kind":"Gdef"},"TensorBox":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.TensorBox","kind":"Gdef"},"Union":{".class":"SymbolTableNode","cross_ref":"typing.Union","kind":"Gdef"},"V":{".class":"SymbolTableNode","cross_ref":"torch._inductor.virtualized.V","kind":"Gdef"},"WorkspaceArg":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.WorkspaceArg","kind":"Gdef"},"WorkspaceZeroMode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.WorkspaceZeroMode","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"_normalize_idx":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["index","total_length"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_kernel._normalize_idx","name":"_normalize_idx","original_first_arg":"index","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["index","total_length"],"arg_types":["builtins.int","builtins.int"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"_normalize_idx","ret_type":"builtins.int","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"cexpr":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.cexpr","name":"cexpr","setter_type":null,"type":{".class":"CallableType","arg_kinds":[0,5,5],"arg_names":["expr","simplify","p"],"arg_types":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"builtins.bool",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":true,"is_ellipsis_args":false,"name":null,"ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"config":{".class":"SymbolTableNode","cross_ref":"torch._inductor.config","kind":"Gdef"},"do_bench_using_profiling":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.do_bench_using_profiling","kind":"Gdef"},"log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.rocm_kernel.log","name":"log","setter_type":null,"type":"logging.Logger"}},"logging":{".class":"SymbolTableNode","cross_ref":"logging","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\rocm_kernel.py"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/rocm_kernel.meta.json

@@ -0,0 +1 @@
+{"data_mtime":1757959416,"dep_lines":[14,15,16,20,7,12,13,6,8,10,11,3,6,2,4,6,1,1,1,1,1,1,1,1,1],"dep_prios":[5,5,5,25,5,5,5,10,5,5,5,5,20,10,5,20,5,30,30,30,30,30,30,30,30],"dependencies":["torch._inductor.codegen.rocm.rocm_benchmark_request","torch._inductor.codegen.rocm.rocm_template_buffer","torch._inductor.codegen.rocm.rocm_utils","torch._inductor.codegen.rocm.rocm_template","torch._inductor.codegen.cpp_wrapper_cpu","torch._inductor.codegen.common","torch._inductor.codegen.cpp_utils","torch._inductor.config","torch._inductor.utils","torch._inductor.ir","torch._inductor.virtualized","collections.abc","torch._inductor","logging","typing","torch","builtins","_frozen_importlib","abc","torch._C","torch._inductor.autotune_process","torch._inductor.ops_handler","torch.utils","torch.utils._sympy","torch.utils._sympy.printers"],"hash":"cc8e9cf9e1eadf49b15ef233cd39a3b1845d38ff","id":"torch._inductor.codegen.rocm.rocm_kernel","ignore_all":true,"interface_hash":"765f476116800b4cb535c8cc8491ed1a8ec4dd86","mtime":1757363946,"options":{"allow_redefinition":false,"allow_redefinition_new":false,"allow_untyped_globals":false,"always_false":[],"always_true":[],"bazel":false,"check_untyped_defs":false,"disable_bytearray_promotion":false,"disable_error_code":[],"disable_memoryview_promotion":false,"disabled_error_codes":[],"disallow_any_decorated":false,"disallow_any_explicit":false,"disallow_any_expr":false,"disallow_any_generics":false,"disallow_any_unimported":false,"disallow_incomplete_defs":false,"disallow_subclassing_any":false,"disallow_untyped_calls":false,"disallow_untyped_decorators":false,"disallow_untyped_defs":false,"enable_error_code":[],"enabled_error_codes":[],"extra_checks":false,"fixed_format_cache":false,"follow_imports":"normal","follow_imports_for_stubs":false,"follow_untyped_imports":false,"ignore_errors":false,"ignore_missing_imports":false,"implicit_optional":false,"implicit_reexport":true,"local_partial_types":false,"mypyc":false,"old_type_inference":false,"platform":"win32","plugins":[],"strict_bytes":false,"strict_concatenate":false,"strict_equality":false,"strict_equality_for_none":false,"strict_optional":true,"warn_no_return":true,"warn_return_any":false,"warn_unreachable":false,"warn_unused_ignores":false},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\rocm_kernel.py","plugin_data":null,"size":10662,"suppressed":[],"version_id":"1.18.1"}

.mypy_cache/3.13/torch/_inductor/codegen/rocm/rocm_template.data.json

@@ -0,0 +1 @@
+{".class":"MypyFile","_fullname":"torch._inductor.codegen.rocm.rocm_template","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","Any":{".class":"SymbolTableNode","cross_ref":"typing.Any","kind":"Gdef"},"ArgInfo":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["builtins.object"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo","name":"ArgInfo","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo","has_param_spec_type":false,"metaclass_type":null,"metadata":{"dataclass":{"attributes":[{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":27,"name":"name","type":"builtins.str"},{"alias":null,"column":4,"has_default":false,"is_in_init":true,"is_init_var":false,"is_neither_frozen_nor_nonfrozen":false,"kw_only":false,"line":28,"name":"ty","type":"builtins.str"}],"frozen":true},"dataclass_tag":{}},"module_name":"torch._inductor.codegen.rocm.rocm_template","mro":["torch._inductor.codegen.rocm.rocm_template.ArgInfo","builtins.object"],"names":{".class":"SymbolTable","__dataclass_fields__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_classvar","is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo.__dataclass_fields__","name":"__dataclass_fields__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"dataclasses.Field"}],"extra_attrs":null,"type_ref":"builtins.dict"}},"plugin_generated":true},"__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0],"arg_names":["self","name","ty"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0],"arg_names":["self","name","ty"],"arg_types":["torch._inductor.codegen.rocm.rocm_template.ArgInfo","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of ArgInfo","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"__match_args__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_ready","allow_incompatible_override"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo.__match_args__","name":"__match_args__","setter_type":null,"type":{".class":"TupleType","implicit":false,"items":[{".class":"LiteralType","fallback":"builtins.str","value":"name"},{".class":"LiteralType","fallback":"builtins.str","value":"ty"}],"partial_fallback":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.tuple"}}},"plugin_generated":true},"__mypy-replace":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[5,5],"arg_names":["name","ty"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_static","is_decorated"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo.__mypy-replace","name":"__mypy-replace","original_first_arg":"name","type":{".class":"CallableType","arg_kinds":[5,5],"arg_names":["name","ty"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-replace of ArgInfo","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_staticmethod","is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo.__mypy-replace","name":"__mypy-replace","setter_type":null,"type":{".class":"CallableType","arg_kinds":[5,5],"arg_names":["name","ty"],"arg_types":["builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__mypy-replace of ArgInfo","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"plugin_generated":true},"__replace__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,5,5],"arg_names":["self","name","ty"],"dataclass_transform_spec":null,"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo.__replace__","name":"__replace__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,5,5],"arg_names":["self","name","ty"],"arg_types":["torch._inductor.codegen.rocm.rocm_template.ArgInfo","builtins.str","builtins.str"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__replace__ of ArgInfo","ret_type":"torch._inductor.codegen.rocm.rocm_template.ArgInfo","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}},"plugin_generated":true},"name":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_property","is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo.name","name":"name","setter_type":null,"type":"builtins.str"}},"ty":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_property","is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo.ty","name":"ty","setter_type":null,"type":"builtins.str"}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.rocm_template.ArgInfo.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.rocm_template.ArgInfo","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"Buffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Buffer","kind":"Gdef"},"DTYPE_TO_ROCM_TYPE":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_utils.DTYPE_TO_ROCM_TYPE","kind":"Gdef"},"IRNode":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.IRNode","kind":"Gdef"},"IndentedBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.IndentedBuffer","kind":"Gdef"},"KernelTemplate":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.common.KernelTemplate","kind":"Gdef"},"Layout":{".class":"SymbolTableNode","cross_ref":"torch._inductor.ir.Layout","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"ROCmBenchmarkRequest":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_benchmark_request.ROCmBenchmarkRequest","kind":"Gdef"},"ROCmTemplate":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"TypeInfo","_promote":[],"abstract_attributes":[],"alt_promote":null,"bases":["torch._inductor.codegen.common.KernelTemplate"],"dataclass_transform_spec":null,"declared_metaclass":null,"defn":{".class":"ClassDef","fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","name":"ROCmTemplate","type_vars":[]},"deletable_attributes":[],"deprecated":null,"flags":[],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","has_param_spec_type":false,"metaclass_type":null,"metadata":{},"module_name":"torch._inductor.codegen.rocm.rocm_template","mro":["torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","torch._inductor.codegen.common.KernelTemplate","builtins.object"],"names":{".class":"SymbolTable","__init__":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0,0,0,1],"arg_names":["self","name","input_nodes","layout","input_reorder"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.__init__","name":"__init__","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,0,0,0,1],"arg_names":["self","name","input_nodes","layout","input_reorder"],"arg_types":["torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","builtins.str",{".class":"Instance","args":["torch._inductor.ir.Buffer"],"extra_attrs":null,"type_ref":"builtins.list"},"torch._inductor.ir.Layout",{".class":"UnionType","items":[{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"__init__ of ROCmTemplate","ret_type":{".class":"NoneType"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"generate":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,4],"arg_names":["self","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.generate","name":"generate","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,4],"arg_names":["self","kwargs"],"arg_types":["torch._inductor.codegen.rocm.rocm_template.ROCmTemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"generate of ROCmTemplate","ret_type":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_runtime_arg_info":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.get_runtime_arg_info","name":"get_runtime_arg_info","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.rocm_template.ROCmTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_info of ROCmTemplate","ret_type":{".class":"Instance","args":["torch._inductor.codegen.rocm.rocm_template.ArgInfo"],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"get_runtime_arg_values":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,4],"arg_names":["self","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.get_runtime_arg_values","name":"get_runtime_arg_values","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,4],"arg_names":["self","kwargs"],"arg_types":["torch._inductor.codegen.rocm.rocm_template.ROCmTemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"get_runtime_arg_values of ROCmTemplate","ret_type":{".class":"Instance","args":[{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":2}],"extra_attrs":null,"type_ref":"builtins.list"},"type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"gfx9_threads_per_warp":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.gfx9_threads_per_warp","name":"gfx9_threads_per_warp","setter_type":null,"type":"builtins.int"}},"globals":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.globals","name":"globals","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.rocm_template.ROCmTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"globals of ROCmTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"header":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["self"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.header","name":"header","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["self"],"arg_types":["torch._inductor.codegen.rocm.rocm_template.ROCmTemplate"],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"header of ROCmTemplate","ret_type":"torch._inductor.utils.IndentedBuffer","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}},"index_counter":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"Var","flags":["is_initialized_in_class","is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.index_counter","name":"index_counter","setter_type":null,"type":{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"itertools.count"}}},"input_nodes":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.input_nodes","name":"input_nodes","setter_type":null,"type":{".class":"Instance","args":["torch._inductor.ir.Buffer"],"extra_attrs":null,"type_ref":"builtins.list"}}},"input_reorder":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.input_reorder","name":"input_reorder","setter_type":null,"type":{".class":"UnionType","items":[{".class":"Instance","args":["builtins.int"],"extra_attrs":null,"type_ref":"builtins.list"},{".class":"NoneType"}],"uses_pep604_syntax":false}}},"layout":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.layout","name":"layout","setter_type":null,"type":"torch._inductor.ir.Layout"}},"output_node":{".class":"SymbolTableNode","implicit":true,"kind":"Mdef","node":{".class":"Var","flags":["explicit_self_type","is_ready","is_inferred"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.output_node","name":"output_node","setter_type":null,"type":"torch._inductor.ir.Buffer"}},"render":{".class":"SymbolTableNode","kind":"Mdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,4],"arg_names":["self","kwargs"],"dataclass_transform_spec":null,"deprecated":null,"flags":["is_trivial_self"],"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.render","name":"render","original_first_arg":"self","type":{".class":"CallableType","arg_kinds":[0,4],"arg_names":["self","kwargs"],"arg_types":["torch._inductor.codegen.rocm.rocm_template.ROCmTemplate",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1}],"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_bound":false,"is_ellipsis_args":false,"name":"render of ROCmTemplate","ret_type":"builtins.str","type_guard":null,"type_is":null,"unpack_kwargs":false,"variables":[]}}}},"self_type":{".class":"TypeVarType","default":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":4},"fullname":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate.Self","id":0,"name":"Self","namespace":"","upper_bound":"torch._inductor.codegen.rocm.rocm_template.ROCmTemplate","values":[],"variance":0},"slots":null,"tuple_type":null,"type_vars":[],"typeddict_type":null}},"ROCmTemplateBuffer":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_template_buffer.ROCmTemplateBuffer","kind":"Gdef"},"ROCmTemplateCaller":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateCaller","kind":"Gdef"},"ROCmTemplateKernel":{".class":"SymbolTableNode","cross_ref":"torch._inductor.codegen.rocm.rocm_kernel.ROCmTemplateKernel","kind":"Gdef"},"Sequence":{".class":"SymbolTableNode","cross_ref":"typing.Sequence","kind":"Gdef"},"TensorMeta":{".class":"SymbolTableNode","cross_ref":"torch._inductor.autotune_process.TensorMeta","kind":"Gdef"},"V":{".class":"SymbolTableNode","cross_ref":"torch._inductor.virtualized.V","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_template.__annotations__","name":"__annotations__","setter_type":null,"type":{".class":"Instance","args":["builtins.str",{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":6}],"extra_attrs":null,"type_ref":"builtins.dict"}}},"__doc__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_template.__doc__","name":"__doc__","setter_type":null,"type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_template.__file__","name":"__file__","setter_type":null,"type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_template.__name__","name":"__name__","setter_type":null,"type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_template.__package__","name":"__package__","setter_type":null,"type":"builtins.str"}},"__spec__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._inductor.codegen.rocm.rocm_template.__spec__","name":"__spec__","setter_type":null,"type":"_frozen_importlib.ModuleSpec"}},"dataclass":{".class":"SymbolTableNode","cross_ref":"dataclasses.dataclass","kind":"Gdef"},"functools":{".class":"SymbolTableNode","cross_ref":"functools","kind":"Gdef"},"itertools":{".class":"SymbolTableNode","cross_ref":"itertools","kind":"Gdef"},"log":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready","is_inferred","has_explicit_value"],"fullname":"torch._inductor.codegen.rocm.rocm_template.log","name":"log","setter_type":null,"type":"logging.Logger"}},"logging":{".class":"SymbolTableNode","cross_ref":"logging","kind":"Gdef"},"patch":{".class":"SymbolTableNode","cross_ref":"unittest.mock.patch","kind":"Gdef"},"unique":{".class":"SymbolTableNode","cross_ref":"torch._inductor.utils.unique","kind":"Gdef"}},"path":"C:\\Users\\marooc\\AppData\\Local\\Programs\\Python\\Python313\\Lib\\site-packages\\torch\\_inductor\\codegen\\rocm\\rocm_template.py"}