TI_MAKE_THREAD_LOCAL

taichi-dev · Jul 24, 2020 · bff3087 · bff3087
1 parent 1fba663
commit bff3087
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 3 deletions.
diff --git a/taichi/backends/metal/codegen_metal.cpp b/taichi/backends/metal/codegen_metal.cpp
@@ -1232,7 +1232,7 @@ FunctionType CodeGen::compile() {
                               /*vectorize=*/false, kernel_->grad,
                               /*ad_use_stack=*/true, config.print_ir,
                               /*lower_global_access=*/true,
-                              /*make_thread_local=*/true);
+                              /*make_thread_local=*/config.make_thread_local);
 
   KernelCodegen codegen(
       taichi_kernel_name_, kernel_->program.snode_root->node_type_name, kernel_,

diff --git a/taichi/backends/opengl/codegen_opengl.cpp b/taichi/backends/opengl/codegen_opengl.cpp
@@ -1007,7 +1007,7 @@ void OpenglCodeGen::lower() {
                               /*vectorize=*/false, kernel_->grad,
                               /*ad_use_stack=*/false, config.print_ir,
                               /*lower_global_access=*/true,
-                              /*make_thread_local=*/true);
+                              /*make_thread_local=*/config.make_thread_local);
 #ifdef _GLSL_DEBUG
   irpass::print(ir);
 #endif

diff --git a/taichi/program/compile_config.cpp b/taichi/program/compile_config.cpp
@@ -35,6 +35,7 @@ CompileConfig::CompileConfig() {
   fast_math = true;
   async_mode = false;
   flatten_if = false;
+  make_thread_local = true;
 
 #if defined(TI_PLATFORM_WINDOWS) or defined(TI_ARCH_ARM)
   use_unified_memory = false;

diff --git a/taichi/program/compile_config.h b/taichi/program/compile_config.h
@@ -35,6 +35,7 @@ struct CompileConfig {
   bool use_unified_memory;
   bool async_mode;
   bool flatten_if;
+  bool make_thread_local;
   DataType default_fp;
   DataType default_ip;
   std::string extra_flags;

diff --git a/taichi/python/export_lang.cpp b/taichi/python/export_lang.cpp
@@ -115,7 +115,8 @@ void export_lang(py::module &m) {
                      &CompileConfig::advanced_optimization)
       .def_readwrite("ad_stack_size", &CompileConfig::ad_stack_size)
       .def_readwrite("async_mode", &CompileConfig::async_mode)
-      .def_readwrite("flatten_if", &CompileConfig::flatten_if);
+      .def_readwrite("flatten_if", &CompileConfig::flatten_if)
+      .def_readwrite("make_thread_local", &CompileConfig::make_thread_local);
 
   m.def("reset_default_compile_config",
         [&]() { default_compile_config = CompileConfig(); });