removerd precision back

tenstorrent · Dec 19, 2024 · 069389e · 069389e
1 parent 80eabbb
commit 069389e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tt-train/sources/ttml/ops/layernorm_op.cpp b/tt-train/sources/ttml/ops/layernorm_op.cpp
@@ -40,7 +40,7 @@ autograd::TensorPtr layernorm(
         mean,
         rstd,
         /* memory_config */ std::nullopt,
-        /* compute_kernel_config */ core::ComputeKernelConfig::precise());
+        /* compute_kernel_config */ std::nullopt);
 
     auto out = autograd::create_tensor();
     out->set_value(out_tensors[0].value());
@@ -63,7 +63,7 @@ autograd::TensorPtr layernorm(
             gamma_grad,
             beta_grad,
             /* memory_config */ std::nullopt,
-            /* compute_kernel_config */ core::ComputeKernelConfig::precise());
+            /* compute_kernel_config */ std::nullopt);
 
         tensor->add_grad(res[0].value());
         gamma->add_grad(res[1].value());