* disable memorize for a while.

apache · Mar 22, 2019 · fc584ce · fc584ce
1 parent 47e83b3
commit fc584ce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/topi/tests/python/test_topi_depthwise_conv2d.py b/topi/tests/python/test_topi_depthwise_conv2d.py
@@ -60,7 +60,7 @@ def check_device(device):
         scale_shift_shape = get_const_tuple(ScaleShift.shape)
 
         # Use memoize, pickle the test data for next time use.
-        @memoize("topi.tests.test_topi_depthwise_conv2d.nchw")
+        #@memoize("topi.tests.test_topi_depthwise_conv2d.nchw")
         def get_ref_data():
             input_np = np.random.uniform(size=input_shape).astype(dtype)
             filter_np = np.random.uniform(size=filter_shape).astype(dtype)
@@ -157,7 +157,7 @@ def check_device(device):
         scale_shift_shape = get_const_tuple(ScaleShift.shape)
 
         # Use memoize, pickle the test data for next time use.
-        @memoize("topi.tests.test_topi_depthwise_conv2d.nhwc.v2")
+        #@memoize("topi.tests.test_topi_depthwise_conv2d.nhwc.v2")
         def get_ref_data():
             input_np = np.random.uniform(size=input_shape).astype(dtype)
             filter_np = np.random.uniform(size=filter_shape).astype(dtype)
@@ -280,7 +280,7 @@ def check_device(device):
         filter_shape = (filter_channel, channel_multiplier, filter_height, filter_width)
 
         # Use memoize, pickle the test data for next time use.
-        @memoize("topi.tests.test_topi_depthwise_conv2d.NCHWc")
+        #@memoize("topi.tests.test_topi_depthwise_conv2d.NCHWc")
         def get_ref_data():
             input_np = np.random.uniform(size=input_shape).astype(dtype)
             filter_np = np.random.uniform(size=filter_shape).astype(dtype)