cuszp compressor import optional

danlkv · May 9, 2024 · 82fb586 · 82fb586
1 parent 1506748
commit 82fb586
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qtensor/compression/Compressor.py b/qtensor/compression/Compressor.py
@@ -17,8 +17,8 @@
 
 
 import torch
-import cuszp
 try:
+    import cuszp
     from cuszx_wrapper import cuszx_host_compress, cuszx_host_decompress, cuszx_device_compress, cuszx_device_decompress
     from cuSZp_wrapper import cuszp_device_compress, cuszp_device_decompress
     from cusz_wrapper import cusz_device_compress, cusz_device_decompress