Skip to content

Commit 7b796cb

Browse files
committed
gpu: nvidia: Added missing bf16 threshold for softmax
1 parent a760af4 commit 7b796cb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/benchdnn/softmax/softmax.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@ void setup_cmp(compare::compare_t &cmp, const prb_t *prb, data_kind_t kind,
232232
const float trh_coeff_bwd = (prb->dir & FLAG_FWD) ? 1.f : 4.f;
233233
const float trh_f32 = trh_coeff_log * trh_coeff_bwd * trh_coeff_f32
234234
* epsilon_dt(trh_dt);
235-
#if DNNL_AARCH64 || defined(DNNL_SYCL_HIP)
235+
#if DNNL_AARCH64 || defined(DNNL_SYCL_HIP) || defined(DNNL_SYCL_CUDA)
236236
// MIOpen and ACL softmax accumulate in F16, but oneDNN now expects accumulation in
237237
// F32, this partially reverts 6727bbe8. For more information on ACL softmax, see
238238
// https://github.com/oneapi-src/oneDNN/issues/1819

0 commit comments

Comments
 (0)