More builtin library support in torch backend. (#2760)

Co-authored-by: Yong He <yhe@nvidia.com>

1 files changed, 3 insertions, 0 deletions

diff --git a/source/slang/slang-stdlib.cpp b/source/slang/slang-stdlib.cpp
index 17fb3d28b..9d4a079c8 100644
--- a/source/slang/slang-stdlib.cpp
+++ b/source/slang/slang-stdlib.cpp
@@ -244,6 +244,9 @@ namespace Slang
         {kIROp_Leq, "leq", "<=", "__BuiltinArithmeticType", ARITHMETIC_MASK | BOOL_RESULT},
     };
 
+    // Integer types that can be used in atomic operations in CUDA.
+    static const char* kCudaAtomicIntegerTypes[] = { "int", "uint", "uint64_t", "int64_t" };
+
     // Both the following functions use these macros.
     // NOTE! They require a variable named path to emit the #line correctly if in source file.
 #define SLANG_RAW(TEXT) sb << TEXT;