Skip to content

Commit b764434

Browse files
deepeshfujitsuvpirogov
authored andcommittedJun 24, 2024
Update softmax.cpp
1 parent 651cf33 commit b764434

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed
 

‎tests/benchdnn/softmax/softmax.cpp

+2
Original file line numberDiff line numberDiff line change
@@ -236,6 +236,8 @@ void setup_cmp(compare::compare_t &cmp, const prb_t *prb, data_kind_t kind,
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
239+
// Similarly, for bf16 on AArch64, the relaxed threshold is necessary due to
240+
// minor accuracy drops observed compared to f32
239241
const float trh = trh_f32;
240242
#else
241243
const bool is_strict_acc

0 commit comments

Comments
 (0)