From 917416f6db7056cddff9d2a0e4e9b4117359157d Mon Sep 17 00:00:00 2001 From: Yong He Date: Thu, 30 Mar 2023 12:50:02 -0700 Subject: More builtin library support in torch backend. (#2760) Co-authored-by: Yong He --- source/slang/slang-stdlib.cpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'source/slang/slang-stdlib.cpp') 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; -- cgit v1.2.3