#13646: Fix CI failure issue

tenstorrent · Oct 25, 2024 · cecb29d · cecb29d
1 parent c16cfb7
commit cecb29d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/tt_eager/python_api_testing/sweep_tests/pytests/tt_dnn/test_broadcast.py b/tests/tt_eager/python_api_testing/sweep_tests/pytests/tt_dnn/test_broadcast.py
@@ -27,7 +27,7 @@
 @pytest.mark.parametrize("dtype", (ttnn.bfloat16, ttnn.bfloat8_b))
 def test_run_bcast_h_test(input_shapes, bcast_op_type, dtype, device, function_level_defaults):
     datagen_func = [
-        generation_funcs.gen_func_with_cast(partial(generation_funcs.gen_rand, low=-100, high=100), torch.float32)
+        generation_funcs.gen_func_with_cast(partial(generation_funcs.gen_rand, low=-100, high=100), torch.bfloat16)
     ] * 2
     comparison_func = partial(comparison_funcs.comp_pcc)
     run_single_pytorch_test(
@@ -60,7 +60,7 @@ def test_run_bcast_h_test(input_shapes, bcast_op_type, dtype, device, function_l
 @pytest.mark.parametrize("dtype", (ttnn.bfloat16, ttnn.bfloat8_b))
 def test_run_bcast_w_test(input_shapes, bcast_op_type, dtype, device, function_level_defaults):
     datagen_func = [
-        generation_funcs.gen_func_with_cast(partial(generation_funcs.gen_rand, low=-100, high=100), torch.float32)
+        generation_funcs.gen_func_with_cast(partial(generation_funcs.gen_rand, low=-100, high=100), torch.bfloat16)
     ] * 2
     comparison_func = partial(comparison_funcs.comp_pcc)
     run_single_pytorch_test(
@@ -93,7 +93,7 @@ def test_run_bcast_w_test(input_shapes, bcast_op_type, dtype, device, function_l
 @pytest.mark.parametrize("dtype", (ttnn.bfloat16, ttnn.bfloat8_b))
 def test_run_bcast_hw_test(input_shapes, bcast_op_type, dtype, device, function_level_defaults):
     datagen_func = [
-        generation_funcs.gen_func_with_cast(partial(generation_funcs.gen_rand, low=-100, high=100), torch.float32)
+        generation_funcs.gen_func_with_cast(partial(generation_funcs.gen_rand, low=-100, high=100), torch.bfloat16)
     ] * 2
     comparison_func = partial(comparison_funcs.comp_pcc)
     run_single_pytorch_test(