Skip to content

Commit 03550b5

Browse files
committed
Set benchmarks to single thread
1 parent 175b78b commit 03550b5

4 files changed

Lines changed: 22 additions & 0 deletions

File tree

example/python_pkg/benchmarks/create_gdfs.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,9 @@
11
import os
2+
nthreads = 1
3+
os.environ["OMP_NUM_THREADS"] = str(nthreads)
4+
os.environ["OPENBLAS_NUM_THREADS"] = str(nthreads)
5+
os.environ["MKL_NUM_THREADS"] = str(nthreads)
6+
27
import glob
38
from raffle.generator import raffle_generator
49
import pytest

example/python_pkg/benchmarks/cutoff_max.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,9 @@
11
import os
2+
nthreads = 1
3+
os.environ["OMP_NUM_THREADS"] = str(nthreads)
4+
os.environ["OPENBLAS_NUM_THREADS"] = str(nthreads)
5+
os.environ["MKL_NUM_THREADS"] = str(nthreads)
6+
27
import pytest
38
import time
49
import numpy as np

example/python_pkg/benchmarks/min_placement.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,9 @@
1+
import os
2+
nthreads = 1
3+
os.environ["OMP_NUM_THREADS"] = str(nthreads)
4+
os.environ["OPENBLAS_NUM_THREADS"] = str(nthreads)
5+
os.environ["MKL_NUM_THREADS"] = str(nthreads)
6+
17
import pytest
28
import time
39
import numpy as np

example/python_pkg/benchmarks/placement_methods.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,9 @@
1+
import os
2+
nthreads = 1
3+
os.environ["OMP_NUM_THREADS"] = str(nthreads)
4+
os.environ["OPENBLAS_NUM_THREADS"] = str(nthreads)
5+
os.environ["MKL_NUM_THREADS"] = str(nthreads)
6+
17
import pytest
28
import time
39
import numpy as np

0 commit comments

Comments
 (0)