[aot] Rename device capability atomic_i64 to atomic_int64 for consist…

…ency (taichi-dev#7095)
lin-hitonami · Jan 12, 2023 · 6c87b57 · 6c87b57
1 parent 6c75400
commit 6c87b57
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/c_api/include/taichi/cpp/taichi.hpp b/c_api/include/taichi/cpp/taichi.hpp
@@ -771,8 +771,8 @@ class CapabilityLevelConfigBuilder {
         value ? TI_TRUE : TI_FALSE;
     return *this;
   }
-  Self &spirv_has_atomic_i64(bool value = true) {
-    cap_level_infos[TI_CAPABILITY_SPIRV_HAS_ATOMIC_I64] =
+  Self &spirv_has_atomic_int64(bool value = true) {
+    cap_level_infos[TI_CAPABILITY_SPIRV_HAS_ATOMIC_INT64] =
         value ? TI_TRUE : TI_FALSE;
     return *this;
   }

diff --git a/c_api/include/taichi/taichi_core.h b/c_api/include/taichi/taichi_core.h
@@ -386,7 +386,7 @@ typedef enum TiCapability {
   TI_CAPABILITY_SPIRV_HAS_INT64 = 4,
   TI_CAPABILITY_SPIRV_HAS_FLOAT16 = 5,
   TI_CAPABILITY_SPIRV_HAS_FLOAT64 = 6,
-  TI_CAPABILITY_SPIRV_HAS_ATOMIC_I64 = 7,
+  TI_CAPABILITY_SPIRV_HAS_ATOMIC_INT64 = 7,
   TI_CAPABILITY_SPIRV_HAS_ATOMIC_FLOAT16 = 8,
   TI_CAPABILITY_SPIRV_HAS_ATOMIC_FLOAT16_ADD = 9,
   TI_CAPABILITY_SPIRV_HAS_ATOMIC_FLOAT16_MINMAX = 10,

diff --git a/docs/lang/articles/c-api/taichi_core.md b/docs/lang/articles/c-api/taichi_core.md
@@ -409,7 +409,7 @@ typedef enum TiCapability {
   TI_CAPABILITY_SPIRV_HAS_INT64 = 4,
   TI_CAPABILITY_SPIRV_HAS_FLOAT16 = 5,
   TI_CAPABILITY_SPIRV_HAS_FLOAT64 = 6,
-  TI_CAPABILITY_SPIRV_HAS_ATOMIC_I64 = 7,
+  TI_CAPABILITY_SPIRV_HAS_ATOMIC_INT64 = 7,
   TI_CAPABILITY_SPIRV_HAS_ATOMIC_FLOAT16 = 8,
   TI_CAPABILITY_SPIRV_HAS_ATOMIC_FLOAT16_ADD = 9,
   TI_CAPABILITY_SPIRV_HAS_ATOMIC_FLOAT16_MINMAX = 10,

diff --git a/python/taichi/lang/enums.py b/python/taichi/lang/enums.py
@@ -15,7 +15,7 @@ class DeviceCapability:
     spirv_has_int64 = "spirv_has_int64"
     spirv_has_float16 = "spirv_has_float16"
     spirv_has_float64 = "spirv_has_float64"
-    spirv_has_atomic_i64 = "spirv_has_atomic_i64"
+    spirv_has_atomic_int64 = "spirv_has_atomic_int64"
     spirv_has_atomic_float16 = "spirv_has_atomic_float16"
     spirv_has_atomic_float16_add = "spirv_has_atomic_float16_add"
     spirv_has_atomic_float16_minmax = "spirv_has_atomic_float16_minmax"

diff --git a/taichi/inc/rhi_constants.inc.h b/taichi/inc/rhi_constants.inc.h
@@ -14,7 +14,7 @@ PER_DEVICE_CAPABILITY(spirv_has_int16)
 PER_DEVICE_CAPABILITY(spirv_has_int64)
 PER_DEVICE_CAPABILITY(spirv_has_float16)
 PER_DEVICE_CAPABILITY(spirv_has_float64)
-PER_DEVICE_CAPABILITY(spirv_has_atomic_i64)
+PER_DEVICE_CAPABILITY(spirv_has_atomic_int64)
 PER_DEVICE_CAPABILITY(spirv_has_atomic_float16)  // load, store, exchange
 PER_DEVICE_CAPABILITY(spirv_has_atomic_float16_add)
 PER_DEVICE_CAPABILITY(spirv_has_atomic_float16_minmax)