@@ -2162,7 +2162,7 @@ static void aclnn_cache_init(ggml_backend_cann_context& ctx, ggml_tensor* dst,
2162
2162
2163
2163
GGML_TENSOR_BINARY_OP_LOCALS
2164
2164
2165
- // theta_scale arange, [0,1,...,ne0/2]
2165
+ // theta_scale arange, [0,1,...,ne0/2]
2166
2166
int64_t theta_scale_length = ne00 / 2 ;
2167
2167
ggml_cann_pool_alloc theta_scale_allocator (ctx.pool (),
2168
2168
theta_scale_length * sizeof (float_t ));
@@ -2171,30 +2171,30 @@ static void aclnn_cache_init(ggml_backend_cann_context& ctx, ggml_tensor* dst,
2171
2171
size_t theta_scale_nb[] = {sizeof (float_t ), sizeof (float_t ), sizeof (float_t ),
2172
2172
theta_scale_length * sizeof (float_t )};
2173
2173
2174
- aclTensor* acl_theat_scale_tensor =
2174
+ aclTensor* acl_theta_scale_tensor =
2175
2175
ggml_cann_create_tensor (theta_scale_buffer, ACL_FLOAT, sizeof (float_t ),
2176
2176
theta_scale_ne, theta_scale_nb, GGML_MAX_DIMS);
2177
2177
float start = 0 ;
2178
2178
float step = 1 ;
2179
2179
float stop = ne00 / 2 ;
2180
2180
float n_elements = ne00 / 2 ;
2181
- aclnn_arange (ctx, acl_theat_scale_tensor , start, stop, step, n_elements);
2181
+ aclnn_arange (ctx, acl_theta_scale_tensor , start, stop, step, n_elements);
2182
2182
2183
2183
// power
2184
2184
aclScalar* acl_theta_scale = aclCreateScalar (&theta_scale, aclDataType::ACL_FLOAT);
2185
- GGML_CANN_CALL_ACLNN_OP (PowScalarTensor, acl_theta_scale, acl_theat_scale_tensor, acl_theat_scale_tensor );
2186
-
2185
+ GGML_CANN_CALL_ACLNN_OP (PowScalarTensor, acl_theta_scale, acl_theta_scale_tensor, acl_theta_scale_tensor );
2186
+
2187
2187
// freq_scale
2188
2188
if (freq_scale != 1 ) {
2189
- aclnn_muls (ctx, acl_theat_scale_tensor , freq_scale, nullptr , true );
2189
+ aclnn_muls (ctx, acl_theta_scale_tensor , freq_scale, nullptr , true );
2190
2190
}
2191
2191
2192
2192
// freq_factors
2193
2193
if (src2) {
2194
2194
aclTensor* acl_freq_factors_tensor = ggml_cann_create_tensor (
2195
2195
src2->data , ggml_cann_type_mapping (src2->type ),
2196
2196
ggml_type_size (src2->type ), theta_scale_ne, theta_scale_nb, GGML_MAX_DIMS);
2197
- aclnn_div (ctx, acl_theat_scale_tensor , acl_freq_factors_tensor);
2197
+ aclnn_div (ctx, acl_theta_scale_tensor , acl_freq_factors_tensor);
2198
2198
ACL_CHECK (aclDestroyTensor (acl_freq_factors_tensor));
2199
2199
}
2200
2200
@@ -2222,7 +2222,7 @@ static void aclnn_cache_init(ggml_backend_cann_context& ctx, ggml_tensor* dst,
2222
2222
aclTensor* acl_theta_tensor =
2223
2223
ggml_cann_create_tensor (theta_buffer, ACL_FLOAT, sizeof (float_t ),
2224
2224
theta_ne, theta_nb, GGML_MAX_DIMS);
2225
- aclnn_mul (ctx, acl_position_tensor, acl_theat_scale_tensor ,
2225
+ aclnn_mul (ctx, acl_position_tensor, acl_theta_scale_tensor ,
2226
2226
acl_theta_tensor);
2227
2227
2228
2228
// sin/cos
@@ -2264,11 +2264,12 @@ static void aclnn_cache_init(ggml_backend_cann_context& ctx, ggml_tensor* dst,
2264
2264
}
2265
2265
2266
2266
// release
2267
- ACL_CHECK (aclDestroyTensor (acl_theat_scale_tensor ));
2267
+ ACL_CHECK (aclDestroyTensor (acl_theta_scale_tensor ));
2268
2268
ACL_CHECK (aclDestroyTensor (acl_position_tensor));
2269
2269
ACL_CHECK (aclDestroyTensor (acl_theta_tensor));
2270
2270
ACL_CHECK (aclDestroyTensor (acl_sin_tensor));
2271
2271
ACL_CHECK (aclDestroyTensor (acl_cos_tensor));
2272
+ ACL_CHECK (aclDestroyScalar (acl_theta_scale));
2272
2273
}
2273
2274
2274
2275
#ifdef __cplusplus
0 commit comments