#7571: Fix build issue in WHB0

tenstorrent · May 17, 2024 · b3da889 · b3da889
1 parent fd6379b
commit b3da889
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/...it_tests/grayskull/test_argmax_padding.py → ...sts/pytests/tt_dnn/test_argmax_padding.py b/...it_tests/grayskull/test_argmax_padding.py → ...sts/pytests/tt_dnn/test_argmax_padding.py
diff --git a/tt_eager/tt_numpy/functions.hpp b/tt_eager/tt_numpy/functions.hpp
@@ -296,7 +296,7 @@ static Tensor index_width(
                 value = 0;
                 index = index + (shape[ultimate] -  up_shape[ultimate]);
             }// dim H
-            index = index + ((shape[penultimate] -  up_shape[penultimate]) * TILE_WIDTH);
+            index = index + ((shape[penultimate] -  up_shape[penultimate]) * tt::constants::TILE_WIDTH);
         } //dim c
     }              // dim N
     auto output = Tensor(OwnedStorage{owned_buffer}, shape, data_type, Layout::ROW_MAJOR).to(layout);
@@ -332,7 +332,7 @@ static Tensor index_height(
                 index = index + (shape[ultimate] -  up_shape[ultimate]);
             }      // dim H
             value = 0;
-            index = index + ((shape[penultimate] -  up_shape[penultimate]) * TILE_WIDTH);
+            index = index + ((shape[penultimate] -  up_shape[penultimate]) * tt::constants::TILE_WIDTH);
         }          // dim C
     }              // dim N
     auto output = Tensor(OwnedStorage{owned_buffer}, shape, data_type, Layout::ROW_MAJOR).to(layout);
@@ -367,7 +367,7 @@ static Tensor index_all(
                 }  // dim W
                 index = index + (shape[ultimate] -  up_shape[ultimate]);
             }// dim H
-            index = index + ((shape[penultimate] -  up_shape[penultimate]) * TILE_WIDTH);
+            index = index + ((shape[penultimate] -  up_shape[penultimate]) * tt::constants::TILE_WIDTH);
         }          // dim C
     }              // dim N
     auto output = Tensor(OwnedStorage{owned_buffer}, shape, data_type, Layout::ROW_MAJOR).to(layout);
@@ -555,7 +555,7 @@ static Tensor index_channel(
                 index = index + (shape[ultimate] - up_shape[ultimate]);
             }      // dim H
             value = value + 1;
-            index = index + ((shape[penultimate] -  up_shape[penultimate]) * TILE_WIDTH);
+            index = index + ((shape[penultimate] -  up_shape[penultimate]) * tt::constants::TILE_WIDTH);
         }          // dim C
         value = 0;
     }              // dim N
@@ -591,7 +591,7 @@ static Tensor index_batch(
                 }  // dim W
                 index = index + (shape[ultimate] - up_shape[ultimate]);
             }      // dim H
-            index = index + ((shape[penultimate] -  up_shape[penultimate]) * TILE_WIDTH);
+            index = index + ((shape[penultimate] -  up_shape[penultimate]) * tt::constants::TILE_WIDTH);
         }          // dim C
         value = value + 1;
     }              // dim N