From 069389e96d65fbe39287e058c6644f66df67c1ed Mon Sep 17 00:00:00 2001 From: Denys Makoviichuk Date: Thu, 19 Dec 2024 20:24:14 +0000 Subject: [PATCH] removerd precision back --- tt-train/sources/ttml/ops/layernorm_op.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tt-train/sources/ttml/ops/layernorm_op.cpp b/tt-train/sources/ttml/ops/layernorm_op.cpp index 03a036d561e..b2f66f37f8e 100644 --- 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());