fixed arch names in runtime.cpp

taichi-dev · Feb 22, 2020 · 8e1e54b · 8e1e54b
1 parent 55cff95
commit 8e1e54b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/python/taichi/main.py b/python/taichi/main.py
@@ -40,10 +40,11 @@ def test_python(test_files=(), verbose=False):
         threads = 2
       env_threads = os.environ.get('TI_TEST_THREADS', '')
       if env_threads:
-        threads = env_threads
+        threads = int(env_threads)
         print(f'Following TI_TEST_THREADS to use {threads} testing thread(s)...')
       print(f'Starting {threads} testing thread(s)...')
-      args += ['-n', str(threads)]
+      if threads > 1:
+        args += ['-n', str(threads)]
   return int(pytest.main(args))
 
 

diff --git a/taichi/runtime/runtime.cpp b/taichi/runtime/runtime.cpp
@@ -30,7 +30,7 @@ using parallel_for_type = void (*)(void *thread_pool,
                                    void *context,
                                    void (*func)(void *, int i));
 
-#if defined(__linux__) && !ARCH_cuda && defined(TI_ARCH_x86_64)
+#if defined(__linux__) && !ARCH_cuda && defined(TI_ARCH_x64)
 __asm__(".symver logf,logf@GLIBC_2.2.5");
 __asm__(".symver powf,powf@GLIBC_2.2.5");
 __asm__(".symver expf,expf@GLIBC_2.2.5");
@@ -128,7 +128,7 @@ int abs_i32(int a) {
   }
 }
 
-#if ARCH_x86_64
+#if ARCH_x64
 
 u32 rand_u32() {
   static u32 x = 123456789, y = 362436069, z = 521288629, w = 88675123;