#8593: Fix pow pcc drop issue

tenstorrent · Oct 16, 2024 · ed19ce0 · ed19ce0
1 parent a73be8a
commit ed19ce0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_pow_float.py b/tests/ttnn/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_pow_float.py
@@ -15,7 +15,7 @@
 def run_pow_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config, data_seed, device):
     torch.manual_seed(data_seed)
 
-    x = torch.Tensor(size=input_shape[0]).uniform_(-100, 100).to(torch.bfloat16)
+    x = torch.Tensor(size=input_shape[0]).uniform_(-100, 100)
     y = random.uniform(0, 10)
 
     try:
@@ -33,6 +33,8 @@ def run_pow_tests(input_shape, dtype, dlayout, in_mem_config, output_mem_config,
 
     assert len(tt_result.shape) == len(ref_value.shape)
     assert tt_result.shape == ref_value.shape
+    print(tt_result.dtype)
+    print(ref_value.dtype)
     assert_with_pcc(ref_value, tt_result, 0.99)