add apis for CTCLoss, test=develop - Githubissues

ROCm / hipDNN

A thin wrapper around miOpen and cuDNN

36 stars 15 forks source link

add apis for CTCLoss, test=develop #50

Open qili93 opened 3 years ago

qili93 commented 3 years ago

Add new apis and both implementation of cudnn and miopen:

hipdnnGetConvolutionNdDescriptor
hipdnnCreateCTCLossDescriptor
hipdnnDestroyCTCLossDescriptor
hipdnnGetCTCLossDescriptor
hipdnnSetCTCLossDescriptor
hipdnnGetCTCLossWorkspaceSize
hipdnnCTCLoss

Also fix "hc::half" undefined in ROCM 3.5 with __half in <hip/hip_fp16.h>