Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel
Commits
397a68f2
Commit
397a68f2
authored
Jun 09, 2022
by
carlushuang
Browse files
close cpu bind core by default
parent
07acd53b
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
include/ck/tensor_operation/cpu/grid/gridwise_gemm_avx2.hpp
include/ck/tensor_operation/cpu/grid/gridwise_gemm_avx2.hpp
+1
-1
include/ck/tensor_operation/cpu/grid/gridwise_gemm_bias_activation_add_avx2.hpp
...ation/cpu/grid/gridwise_gemm_bias_activation_add_avx2.hpp
+1
-1
No files found.
include/ck/tensor_operation/cpu/grid/gridwise_gemm_avx2.hpp
View file @
397a68f2
...
@@ -304,7 +304,7 @@ struct GridwiseGemmAvx2_MxN
...
@@ -304,7 +304,7 @@ struct GridwiseGemmAvx2_MxN
int
total_threads
=
omp_get_max_threads
();
int
total_threads
=
omp_get_max_threads
();
if
(
total_threads
>
1
&&
ck
::
getenv_int
(
"CK_CPU_BIND_CORE"
,
1
)
!=
0
)
if
(
total_threads
>
1
&&
ck
::
getenv_int
(
"CK_CPU_BIND_CORE"
,
0
)
!=
0
)
{
{
#pragma omp parallel
#pragma omp parallel
{
{
...
...
include/ck/tensor_operation/cpu/grid/gridwise_gemm_bias_activation_add_avx2.hpp
View file @
397a68f2
...
@@ -330,7 +330,7 @@ struct GridwiseGemmBiasActivationAddAvx2_MxN
...
@@ -330,7 +330,7 @@ struct GridwiseGemmBiasActivationAddAvx2_MxN
int
total_threads
=
omp_get_max_threads
();
int
total_threads
=
omp_get_max_threads
();
if
(
total_threads
>
1
&&
ck
::
getenv_int
(
"CK_CPU_BIND_CORE"
,
1
)
!=
0
)
if
(
total_threads
>
1
&&
ck
::
getenv_int
(
"CK_CPU_BIND_CORE"
,
0
)
!=
0
)
{
{
#pragma omp parallel
#pragma omp parallel
{
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment