Skip to content
This repository was archived by the owner on Aug 30, 2024. It is now read-only.

Commit

Permalink
Enable OMP in BesTLA (#121)
Browse files Browse the repository at this point in the history
  • Loading branch information
airMeng authored Feb 6, 2024
1 parent 08be9a6 commit 3afae42
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,7 @@ if (NS_BTLA_UT)
endif()
include(FindOpenMP)

set(BTLA_USE_OPENMP ON)
set(BTLA_USE_OPENMP ON CACHE BOOL "BesTLA use OpenMP" FORCE)
add_subdirectory(bestla)

add_subdirectory(neural_speed)
2 changes: 1 addition & 1 deletion neural_speed/core/ne_layers.c
Original file line number Diff line number Diff line change
Expand Up @@ -3037,7 +3037,7 @@ struct ne_tensor* ne_rope_impl(struct ne_context* ctx, struct ne_tensor* a, int

ne_scratch_load(ctx);

/* what the diffrence of setting parameters in b->data and in op_parameters */
/* what the difference of setting parameters in b->data and in op_parameters */
/* float and int are in different data ?? */
float params[] = {freq_base, freq_scale, (float)yarn_orig_ctx, ext_factor, attn_factor, beta_fast, beta_slow};
ne_set_op_params(result, &params, sizeof(params));
Expand Down

0 comments on commit 3afae42

Please sign in to comment.