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

Implement CosineEmbeddingLoss

Open hieule88 opened this issue 10 months ago • 0 comments

  • Added CosineEmbeddingLoss forward and backward operation and kernel.
  • Added driver test and gtest for CosineEmbeddingLoss.
  • New API is guarded by MIOPEN_BETA_API macro.
  • Average over all cases:

No Reduction:

type Forward Backward
float16 6.42 4.91
float32 5.46 4.56
bfloat16 7.21 5.04

Reduction:

type Forward Backward
float16 3.41 5.61
float32 2.85 5.12
bfloat16 3.48 5.93

hieule88 avatar Mar 07 '25 04:03 hieule88