"src/nni_manager/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "04d2d7cb1f2c3f279fa314e51351bdf80dc4d374"
Unverified Commit 48e70e68 authored by Lei Wang's avatar Lei Wang Committed by GitHub
Browse files

[Cache] Rename sparse compress cache directory (#1460)

* Enhance cache directory structure by including version information in sparse.py to ensure separate caches for different versions.

* Fix formatting in sparse.py by adding a newline for improved readability and consistency.
parent 91cf7966
...@@ -7,10 +7,13 @@ from tilelang.utils.tensor import is_float8_dtype, fp8_remove_negative_zeros_ ...@@ -7,10 +7,13 @@ from tilelang.utils.tensor import is_float8_dtype, fp8_remove_negative_zeros_
from torch.utils.cpp_extension import load, _import_module_from_library from torch.utils.cpp_extension import load, _import_module_from_library
from tilelang import env from tilelang import env
# Include version information to ensure different versions use separate caches
from tilelang import __version__
# Define paths # Define paths
compress_util = os.path.join(env.TILELANG_TEMPLATE_PATH, "tl_templates/cuda/compress_sm90.cu") compress_util = os.path.join(env.TILELANG_TEMPLATE_PATH, "tl_templates/cuda/compress_sm90.cu")
# Cache directory for compiled extensions # Cache directory for compiled extensions
_CACHE_DIR = os.path.join(env.TILELANG_CACHE_DIR, "sparse_compressor") _CACHE_DIR = os.path.join(env.TILELANG_CACHE_DIR, "sparse_compressor", __version__)
os.makedirs(_CACHE_DIR, exist_ok=True) os.makedirs(_CACHE_DIR, exist_ok=True)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment