-
Couldn't load subscription status.
- Fork 286
Open
Labels
questionFurther information is requestedFurther information is requested
Description
Required prerequisites
- I have read the documentation https://tilelang.com.
- I have searched the Issue Tracker that this hasn't already been reported. (comment there if it has.)
Questions
While func_or_mod is tir.PrimFunc | tvm.IRModule, why accessing __name__ of it? The name of function should inside its attrs
tilelang/tilelang/jit/adapter/torch/metal.py
Lines 16 to 31 in 6e1dc6a
| def __init__( | |
| self, | |
| params: list[KernelParam], | |
| result_idx: list[int], | |
| # target: Union[str, Target], | |
| func_or_mod: tir.PrimFunc | tvm.IRModule, | |
| # host_mod: Optional[tvm.IRModule] = None, | |
| device_mod: tvm.IRModule | None = None, | |
| kernel_global_source: str | None = None, | |
| verbose: bool = False, | |
| # pass_configs: Optional[Dict[str, Any]] = None, | |
| # compile_flags: Optional[List[str]] = None | |
| ): | |
| self.kernel_global_source = kernel_global_source | |
| self.kernel_name = func_or_mod.__name__ + '_kernel' | |
| self.verbose = verbose |
Metadata
Metadata
Assignees
Labels
questionFurther information is requestedFurther information is requested