 Jinjie LiuandZihua 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 |
|