Jinjie Liu and Zihua Wu
9ff7c523a9
[Enhancement] Refactor KernelCache to use inheritance-based design ( #1483 )
...
* refactor kernel_cache
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* set key to auto when execution_backend is None
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* fix type error on _load_kernel_source override
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* put kernel cache subclasses in adapter
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* remove auto in execution_backend
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* remove _save_kernel_source_code_to_disk for NVRTCKernelCache
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* refactor all cutedsl if statements
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* remove unused kernel_cubin_path
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* add func arguments
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* remove ctypes
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* update ci.yaml to test kernel cache
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* fix bugs after #1512 merged
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
* test: fix kernel cache test
* fix: ruff
* fix: ruff
* fix: coderabbit
* fix: coderabbit
* fix: ruff
* fix: MMA
* fix: coderabbit
* fix: codex
* fix: coderabbit
* fix: coderabbit
---------
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn >
Co-authored-by: Zihua Wu <13583761+lucifer1004@users.noreply.github.com >
2025-12-26 15:41:22 +08:00