From ce7994359a536745dd079135e77c2baaf3a059ae Mon Sep 17 00:00:00 2001 From: LeiWang1999 Date: Wed, 7 Aug 2024 07:58:05 +0000 Subject: [PATCH] bug fix --- testing/python/cache/test_operator_cache.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/testing/python/cache/test_operator_cache.py b/testing/python/cache/test_operator_cache.py index 7d609c7c..3f80951e 100644 --- a/testing/python/cache/test_operator_cache.py +++ b/testing/python/cache/test_operator_cache.py @@ -2,6 +2,7 @@ # Licensed under the MIT License. import pytest import os +import shutil import torch import bitblas from bitblas import Matmul, MatmulConfig @@ -248,7 +249,7 @@ def test_global_cache_save_to_database( database_path = "/tmp/.tmp_bitblas_cache.db" # clean the database if exists if os.path.exists(database_path): - os.remove(database_path) + shutil.rmtree(database_path) global_operator_cache.save_into_database(database_path, target=target) assert os.path.exists(database_path) global_operator_cache.clear() @@ -285,6 +286,4 @@ def test_global_cache_save_to_database( # fmt: on if __name__ == "__main__": - # bitblas.testing.main() - test_global_cache_save_to_database(1, 1024, 1024, "float16", "float16", "float16", False, False, - False, "nt", False) + bitblas.testing.main()