Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "97c452823c6cf3cd28e8838cefcceb80da1c1ed7"
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(
switch (_info.data_type) {
case INFINI_DTYPE_F16:
ROPE_TYPE(half);
case INFINI_DTYPE_BF16:
ROPE_TYPE(bfloat16_t);
case INFINI_DTYPE_F32:
ROPE_TYPE(float);
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