Skip to content

Commit

Permalink
adjustment
Browse files Browse the repository at this point in the history
  • Loading branch information
noemotiovon committed Nov 26, 2024
1 parent c71809c commit fe1f1c9
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion ggml/src/ggml-cann/aclnn_ops.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -2948,7 +2948,7 @@ void ggml_cann_rope(ggml_backend_cann_context& ctx, ggml_tensor* dst) {
int acl_mode = mode;
if (mode == 0) {
acl_mode = 1;
}
}

aclTensor* acl_x = ggml_cann_create_tensor(src0);
aclTensor* acl_dst = ggml_cann_create_tensor(dst);
Expand Down
6 changes: 3 additions & 3 deletions ggml/src/ggml-cann/ggml-cann.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1721,7 +1721,7 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev,
case GGML_OP_ROPE: {
// TODO: with ops-test v == 1
float freq_scale, attn_factor, ext_factor;
memcpy(&freq_scale, (int32_t*)op->op_params + 6, sizeof(float));
memcpy(&freq_scale, (int32_t*)op->op_params + 6, sizeof(float));
memcpy(&attn_factor, (int32_t*)op->op_params + 8, sizeof(float));
memcpy(&ext_factor, (int32_t*)op->op_params + 7, sizeof(float));
// TODO: with freq_factors
Expand Down Expand Up @@ -1760,8 +1760,8 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev,
}
return true;
}
case GGML_OP_IM2COL:
case GGML_OP_CONCAT:
case GGML_OP_IM2COL:
case GGML_OP_CONCAT:
case GGML_OP_DUP:
case GGML_OP_REPEAT:
case GGML_OP_NONE:
Expand Down

0 comments on commit fe1f1c9

Please sign in to comment.