diff --git a/examples/gemv/example_gemv.py b/examples/gemv/example_gemv.py index 58e0114be..3772dc6bd 100644 --- a/examples/gemv/example_gemv.py +++ b/examples/gemv/example_gemv.py @@ -360,7 +360,7 @@ def main(do_bench: bool = True): print("Test passed!") - if not do_bench: + if do_bench: best_result = get_autotuned_kernel(N, K) best_config = best_result.config kernel = splitk_gemv_vectorized_tvm(N, K, **best_config) diff --git a/tilelang/utils/sparse.py b/tilelang/utils/sparse.py index 1021852ca..a7b17ad93 100644 --- a/tilelang/utils/sparse.py +++ b/tilelang/utils/sparse.py @@ -16,11 +16,10 @@ def _get_cached_lib(): name = 'compress_lib' - cached_path = os.path.join(_CACHE_DIR, f"{name}.so") - if os.path.exists(cached_path): + if os.path.exists(os.path.join(_CACHE_DIR, f"{name}.so")): try: - return _import_module_from_library(name, cached_path) + return _import_module_from_library(name, _CACHE_DIR, is_python_module=True) except Exception: # If loading fails, recompile pass