[FEATURE] Make up float32 input for bf16/fp16 cpu tracing with Ratex - Githubissues

awslabs / raf

Apache License 2.0

140 stars 20 forks source link

[FEATURE] Make up float32 input for bf16/fp16 cpu tracing with Ratex #117

Closed XinweiFu closed 2 years ago

XinweiFu commented 2 years ago

Description

This PR makes up float32 input for bf16/fp16 cpu tracing with Ratex.

Checklist

[x] PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL], [FEATURE], [DOC], etc)
[x] Changes are complete (i.e. I finished coding on this PR)
[ ] All changes have test coverage
[x] Code is well-documented

cc @awslabs/raf-reviewer