diff options
| author | Yaxun (Sam) Liu <yaxun.liu@amd.com> | 2022-02-21 20:16:04 -0500 |
|---|---|---|
| committer | Yaxun (Sam) Liu <yaxun.liu@amd.com> | 2022-02-23 22:28:29 -0500 |
| commit | 9d899d8f01872e91b9909a6ee5937a796a399276 (patch) | |
| tree | 931251413b16294d593af97ec133cf269eeeb580 /clang/lib/Frontend/InitPreprocessor.cpp | |
| parent | b01430a04f191d2191be98568d721c8f111d9b71 (diff) | |
| download | llvm-9d899d8f01872e91b9909a6ee5937a796a399276.zip llvm-9d899d8f01872e91b9909a6ee5937a796a399276.tar.gz llvm-9d899d8f01872e91b9909a6ee5937a796a399276.tar.bz2 | |
[HIP] Support `-fgpu-default-stream`
Introduce -fgpu-default-stream={legacy|per-thread} option to
support per-thread default stream for HIP runtime.
When -fgpu-default-stream=per-thread, HIP kernels are
launched through hipLaunchKernel_spt instead of
hipLaunchKernel. Also HIP_API_PER_THREAD_DEFAULT_STREAM=1
is defined by the preprocessor to enable other per-thread stream
API's.
Reviewed by: Artem Belevich
Differential Revision: https://reviews.llvm.org/D120298
Diffstat (limited to 'clang/lib/Frontend/InitPreprocessor.cpp')
| -rw-r--r-- | clang/lib/Frontend/InitPreprocessor.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/clang/lib/Frontend/InitPreprocessor.cpp b/clang/lib/Frontend/InitPreprocessor.cpp index ff507e2..6b7c743 100644 --- a/clang/lib/Frontend/InitPreprocessor.cpp +++ b/clang/lib/Frontend/InitPreprocessor.cpp @@ -538,6 +538,9 @@ static void InitializeStandardPredefinedMacros(const TargetInfo &TI, Builder.defineMacro("__HIP_MEMORY_SCOPE_SYSTEM", "5"); if (LangOpts.CUDAIsDevice) Builder.defineMacro("__HIP_DEVICE_COMPILE__"); + if (LangOpts.GPUDefaultStream == + LangOptions::GPUDefaultStreamKind::PerThread) + Builder.defineMacro("HIP_API_PER_THREAD_DEFAULT_STREAM"); } } |