Skip to content

float16 support to non-cudnn softmax op on GPU #9684

@kexinzhao

Description

@kexinzhao

Currently, the default example code use non-cudnn softmax op to generate the prediction result.
We thus need to add float16 support to non-cudnn softmax op on GPU in addition to the cudnn one.

Metadata

Metadata

Assignees

Labels

预测原名Inference,包含Capi预测问题等

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions