#11621: remove bfloat8

tenstorrent · Oct 14, 2024 · aa4c83c · aa4c83c
1 parent 790a284
commit aa4c83c
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/tests/sweep_framework/sweeps/data_movement/permute/permute_pytorch2_tiled.py b/tests/sweep_framework/sweeps/data_movement/permute/permute_pytorch2_tiled.py
@@ -260,7 +260,7 @@
             {"shape": [1, 16, 256, 64], "dims": [0, 2, 1, 3]},
             {"shape": [1, 256, 16, 64], "dims": [0, 2, 1, 3]},
         ],
-        "dtype": [ttnn.bfloat16, ttnn.bfloat8_b],
+        "dtype": [ttnn.bfloat16],
         "layout": [ttnn.TILE_LAYOUT],
     },
 }

diff --git a/tests/sweep_framework/sweeps/data_movement/repeat/repeat_pytorch2.py b/tests/sweep_framework/sweeps/data_movement/repeat/repeat_pytorch2.py
@@ -34,7 +34,7 @@
             {"shape": [6, 2], "repeats": [400, 1]},
             {"shape": [6, 2], "repeats": [9, 1]},
         ],
-        "dtype": [ttnn.bfloat16, ttnn.bfloat8_b],
+        "dtype": [ttnn.bfloat16],
         "layout": [ttnn.ROW_MAJOR_LAYOUT, ttnn.TILE_LAYOUT],
     }
 }

diff --git a/tests/sweep_framework/sweeps/data_movement/slice/slice_pytorch2_tiled.py b/tests/sweep_framework/sweeps/data_movement/slice/slice_pytorch2_tiled.py
@@ -172,7 +172,7 @@
             {"dims": [8732, 4], "dim": 1, "start": 0, "end": -1, "step": 4},
             {"dims": [8732, 4], "dim": 1, "start": 0, "end": 2},
         ],
-        "dtype": [ttnn.bfloat16, ttnn.bfloat8_b],
+        "dtype": [ttnn.bfloat16],
         "layout": [ttnn.TILE_LAYOUT],
     }
 }

diff --git a/tests/sweep_framework/sweeps/data_movement/split/split_pytorch2.py b/tests/sweep_framework/sweeps/data_movement/split/split_pytorch2.py
@@ -33,7 +33,7 @@
             {"shape": [768, 256], "split_size": 256, "dim": -1},
             {"shape": [768], "split_size": 256, "dim": -1},
         ],
-        "dtype": [ttnn.bfloat16, ttnn.bfloat8_b],
+        "dtype": [ttnn.bfloat16],
         "layout": [ttnn.ROW_MAJOR_LAYOUT, ttnn.TILE_LAYOUT],
     }
 }

diff --git a/tests/sweep_framework/sweeps/data_movement/squeeze/squeeze_pytorch2.py b/tests/sweep_framework/sweeps/data_movement/squeeze/squeeze_pytorch2.py
@@ -56,7 +56,7 @@ def generate_squeeze_config(num_samples=10):
             {"shape": [1, 1, 25088], "dim": 0},
             {"shape": [1, 1, 480, 640], "dim": 1},
         ],
-        "dtype": [ttnn.bfloat16, ttnn.bfloat8_b],
+        "dtype": [ttnn.bfloat16],
         "layout": [ttnn.ROW_MAJOR_LAYOUT, ttnn.TILE_LAYOUT],
     }
 }

diff --git a/tests/sweep_framework/sweeps/data_movement/stack/stack_pytorch2.py b/tests/sweep_framework/sweeps/data_movement/stack/stack_pytorch2.py
@@ -66,7 +66,7 @@
             {"tensors_shapes": [[9], [9], [9], [9], [9], [9], [9], [9], [9], [9], [9], [9]], "dim": -1},
             {"tensors_shapes": [[9], [9], [9], [9], [9], [9], [9], [9]], "dim": -1},
         ],
-        "dtype": [ttnn.bfloat16, ttnn.bfloat8_b],
+        "dtype": [ttnn.bfloat16],
         "layout": [ttnn.ROW_MAJOR_LAYOUT, ttnn.TILE_LAYOUT],
     }
 }

diff --git a/tests/sweep_framework/sweeps/data_movement/unsqueeze/unsqueeze_pytorch2.py b/tests/sweep_framework/sweeps/data_movement/unsqueeze/unsqueeze_pytorch2.py
@@ -264,7 +264,7 @@ def generate_unsqueeze_config(num_samples=10):
             {"shape": [1], "dim": 1},
             {"shape": [0, 256], "dim": 0},
         ],
-        "dtype": [ttnn.bfloat16, ttnn.bfloat8_b],
+        "dtype": [ttnn.bfloat16],
         "layout": [ttnn.ROW_MAJOR_LAYOUT, ttnn.TILE_LAYOUT],
     }
 }