diff --git a/src/neural/backends/cuda/network_cuda.cc b/src/neural/backends/cuda/network_cuda.cc index d95310f9b9..fa090a144f 100644 --- a/src/neural/backends/cuda/network_cuda.cc +++ b/src/neural/backends/cuda/network_cuda.cc @@ -863,7 +863,7 @@ class CudaNetwork : public Network { float sum = w + d + l; w /= sum; l /= sum; - d = 1.0f - w - l; + d /= sum; io->op_value_mem_[3 * i + 0] = w; io->op_value_mem_[3 * i + 1] = d; io->op_value_mem_[3 * i + 2] = l; diff --git a/src/neural/backends/cuda/network_cudnn.cc b/src/neural/backends/cuda/network_cudnn.cc index 7b13e8b0b8..8bb2df286e 100644 --- a/src/neural/backends/cuda/network_cudnn.cc +++ b/src/neural/backends/cuda/network_cudnn.cc @@ -879,7 +879,7 @@ class CudnnNetwork : public Network { float sum = w + d + l; w /= sum; l /= sum; - d = 1.0f - w - l; + d /= sum; io->op_value_mem_[3 * i + 0] = w; io->op_value_mem_[3 * i + 1] = d; io->op_value_mem_[3 * i + 2] = l; diff --git a/src/neural/backends/network_onnx.cc b/src/neural/backends/network_onnx.cc index 730e36170a..8877302ecc 100644 --- a/src/neural/backends/network_onnx.cc +++ b/src/neural/backends/network_onnx.cc @@ -313,7 +313,7 @@ void OnnxComputation::ComputeBlocking() { float sum = w + d + l; w /= sum; l /= sum; - d = 1.0f - w - l; + d /= sum; } wdl_output_data_[3 * i + 0] = w; wdl_output_data_[3 * i + 1] = d; diff --git a/src/neural/backends/sycl/network_sycl.cc.dp.cpp b/src/neural/backends/sycl/network_sycl.cc.dp.cpp index ca551f3a7b..11683c8aae 100644 --- a/src/neural/backends/sycl/network_sycl.cc.dp.cpp +++ b/src/neural/backends/sycl/network_sycl.cc.dp.cpp @@ -882,7 +882,7 @@ class SyclNetwork : public Network { float sum = w + d + l; w /= sum; l /= sum; - d = 1.0f - w - l; + d /= sum; io->op_value_mem_shared_[3 * i + 0] = w; io->op_value_mem_shared_[3 * i + 1] = d; io->op_value_mem_shared_[3 * i + 2] = l;