MIOpen icon indicating copy to clipboard operation
MIOpen copied to clipboard
verified publisher icon ROCmSoftwarePlatform

Implement KLDivLoss backward

Open hieule88 opened this issue 10 months ago • 0 comments

  • Added no reduction KLDivLoss backward operation and kernel.

  • Added driver test and gtest for KLDivLoss.

  • New API is guarded by MIOPEN_BETA_API macro.

  • Average over all cases:

No Reduction:

type Backward
float16 2.33
float32 2.05
bfloat16 2.22

Reduction:

type Backward
float16 2.42
float32 2.15
bfloat16 2.32

hieule88 avatar Mar 10 '25 16:03 hieule88