update test target

apache · Mar 4, 2021 · 3961de8 · 3961de8
1 parent 15189ca
commit 3961de8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/topi/python/test_topi_cumsum.py b/tests/python/topi/python/test_topi_cumsum.py
@@ -48,7 +48,7 @@ def check_cumsum(np_ref, data, axis=None, dtype=None):
         check_cumsum(np.cumsum(data, dtype=np.int32), data, dtype="int32")
 
     for in_dtype in ["float32", "float64"]:
-        if str(target.kind) == 'metal' and in_dtype == 'float64':
+        if target == 'metal' and in_dtype == 'float64':
             # float64 is not supported in metal
             continue
         data = np.random.randn(10, 10).astype(in_dtype)