Skip to content

GitLab

Commit e3669dfc authored by wooway777's avatar wooway777
Browse files

issue/265 - added bf16 support for Cambricon RoPE

parent 330d9189
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/infiniop/ops/rope/bang/rope_bang.mlu
View file @ e3669dfc
...@@ -108,6 +108,8 @@ infiniStatus_t Descriptor::calculate( ...@@ -108,6 +108,8 @@ infiniStatus_t Descriptor::calculate(
switch (_info.data_type) { switch (_info.data_type) {
case INFINI_DTYPE_F16: case INFINI_DTYPE_F16:
ROPE_TYPE(half); ROPE_TYPE(half);
case INFINI_DTYPE_BF16:
ROPE_TYPE(bfloat16_t);
case INFINI_DTYPE_F32: case INFINI_DTYPE_F32:
ROPE_TYPE(float); ROPE_TYPE(float);
default: default:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment