fix bug

Deep-Learning-Profiling-Tools · Mar 21, 2024 · eaa5505 · eaa5505
1 parent ff15a02
commit eaa5505
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/triton_viz/draw.py b/triton_viz/draw.py
@@ -237,7 +237,7 @@ def store_load(
     tensor, tensor_id = tensor_table[x.ptr]
     # inp = base_tensor(tensor.shape, DEFAULT)
     color = ACTIVE[tensor_id]
-    invalid = (x.invalid_access_masks).any()
+    invalid = x.invalid_access_masks.any()
     if invalid:
         color = Color("red")
     inp = cover(tensor.shape, tensor.dtype, x.original_offsets, x.original_masks, color)