#3647: Add fix for test for polyval coeffs generation

tenstorrent · Nov 21, 2023 · 1062044 · 1062044
1 parent 64d742f
commit 1062044
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/tt_eager/python_api_testing/sweep_tests/generation_funcs.py b/tests/tt_eager/python_api_testing/sweep_tests/generation_funcs.py
@@ -1324,7 +1324,7 @@ def gen_polyval_args(
     ):
         if input_info is not None:
             num_coeffs = torch.tensor(1, dtype=torch.int).random_(1, max_num_coeffs + 1).item()
-            coeffs = torch.Tensor(num_coeffs).uniform_(low, high).tolist()
+            coeffs = torch.Tensor(num_coeffs).uniform_(low, high).to(torch.bfloat16).tolist()
             input_info.update({"coeffs": coeffs})
             yield input_info