diff options
| author | NBickford <57467222+NBickford-NV@users.noreply.github.com> | 2024-02-24 15:46:08 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-24 15:46:08 -0800 |
| commit | d7bb80673a3ef91ec904ca487dcc71315054c654 (patch) | |
| tree | c5bf6c73252fbc65b3a79c4816f30b5c31b20d1a | |
| parent | a1827ee5e9b8088b23db3fa688b7bd62b7bbe9ac (diff) | |
Enable SLANG_MAKE_VECTOR calls when using SLANG_CUDA_ENABLE_HALF without SLANG_CUDA_RTC (#3624)
| -rw-r--r-- | prelude/slang-cuda-prelude.h | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/prelude/slang-cuda-prelude.h b/prelude/slang-cuda-prelude.h index 38f8a721a..74a9d0cec 100644 --- a/prelude/slang-cuda-prelude.h +++ b/prelude/slang-cuda-prelude.h @@ -363,11 +363,14 @@ SLANG_CUDA_VECTOR_FLOAT_OPS(__half) SLANG_CUDA_FLOAT_VECTOR_MOD(float) SLANG_CUDA_FLOAT_VECTOR_MOD(double) -#if SLANG_CUDA_RTC +#if SLANG_CUDA_RTC || SLANG_CUDA_ENABLE_HALF #define SLANG_MAKE_VECTOR(T) \ SLANG_FORCE_INLINE SLANG_CUDA_CALL T##2 make_##T##2(T x, T y) { return T##2{x, y}; }\ SLANG_FORCE_INLINE SLANG_CUDA_CALL T##3 make_##T##3(T x, T y, T z) { return T##3{ x, y, z }; }\ SLANG_FORCE_INLINE SLANG_CUDA_CALL T##4 make_##T##4(T x, T y, T z, T w) { return T##4{ x, y, z, w }; } +#endif + +#if SLANG_CUDA_RTC SLANG_MAKE_VECTOR(int) SLANG_MAKE_VECTOR(uint) SLANG_MAKE_VECTOR(short) @@ -413,6 +416,9 @@ SLANG_MAKE_VECTOR_FROM_SCALAR(float) SLANG_MAKE_VECTOR_FROM_SCALAR(double) #if SLANG_CUDA_ENABLE_HALF SLANG_MAKE_VECTOR_FROM_SCALAR(__half) +#if !SLANG_CUDA_RTC +SLANG_FORCE_INLINE SLANG_CUDA_CALL __half1 make___half1(__half x) { return __half1{x}; } +#endif #endif #define SLANG_CUDA_VECTOR_ATOMIC_BINARY_IMPL(Fn,T,N) \ |