[Minor] debug for unittest (#259)

mit-han-lab · Nov 7, 2023 · 7c04c58 · 7c04c58
1 parent afa2e3b
commit 7c04c58
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/python/test_single_layer_conv.py b/tests/python/test_single_layer_conv.py
@@ -9,7 +9,7 @@
 from torchsparse.nn import functional as F
 from torchsparse.utils import make_ntuple
 
-from test_utils import *
+from .test_utils import *
 
 __all__ = ["test_single_layer_convolution_forward"]
 

diff --git a/tests/test.py b/tests/test.py
@@ -25,9 +25,9 @@ def test_single_layer(self):
                 count += 1
 
         # switch to hashmap_on_the_fly
-        config = F.get_default_conv_config()
+        config = F.conv_config.get_default_conv_config()
         config.kmap_mode = "hashmap_on_the_fly"
-        F.set_global_conv_config(config)
+        F.conv_config.set_global_conv_config(config)
         for kernel_size in kernel_sizes:
             for stride in strides:
                 mean_adiff, max_rdiff = test_single_layer_convolution_forward(