fix bug

Deep-Learning-Profiling-Tools · Mar 21, 2024 · ff15a02 · ff15a02
1 parent 2ec88ad
commit ff15a02
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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)

diff --git a/triton_viz/interpreter.py b/triton_viz/interpreter.py
@@ -190,7 +190,7 @@ def check_out_of_bounds_access(ptrs, masks):
     offsets = ptrs.data - tensor_ptr.ptr
     max_valid_offset = np.prod(tensor_ptr.shape) * tensor_ptr.element_size
     valid_access_masks = (offsets >= 0) & (offsets < max_valid_offset)
-    invalid_access_masks = (~valid_access_masks) & (~masks.data)
+    invalid_access_masks = (~valid_access_masks) & masks.data
     corrected_offsets = np.where(valid_access_masks, offsets, 0)
     return (
         tensor_ptr,