Commit 30c3a11b authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

ban numpy multithreading

parent 626f8bc7

csst_common/pipeline.py

+3 −0
Original line number Diff line number Diff line
@@ -74,7 +74,10 @@ class Pipeline: 


    def ban_multithreading(self):

        os.environ["OMP_NUM_THREADS"] = "1"

        os.environ["OPENBLAS_NUM_THREADS"] = "1"

        os.environ["MKL_NUM_THREADS"] = "1"

        os.environ["VECLIB_MAXIMUM_THREADS"] = "1"

        os.environ["NUMEXPR_NUM_THREADS"] = "1"



    def set_test_env(self):

        os.environ["CSST_DFS_API_MODE"] = "cluster"