From f46d6cf249710822e256296c772187a12a2f9e97 Mon Sep 17 00:00:00 2001 From: Michael Liao Date: Fri, 8 Mar 2019 09:08:39 -0500 Subject: [PATCH] [hip] Re-implement hipLaunchKernelGGL as macros. --- hipamd/include/hip/hcc_detail/hip_runtime.h | 19 ++++++++----------- 1 file changed, 8 insertions(+), 11 deletions(-) diff --git a/hipamd/include/hip/hcc_detail/hip_runtime.h b/hipamd/include/hip/hcc_detail/hip_runtime.h index 60d145c884..88cdf7e9ba 100644 --- a/hipamd/include/hip/hcc_detail/hip_runtime.h +++ b/hipamd/include/hip/hcc_detail/hip_runtime.h @@ -331,18 +331,15 @@ extern void ihipPostLaunchKernel(const char* kernelName, hipStream_t stream, gri typedef int hipLaunchParm; -template -inline void hipLaunchKernelGGL(F&& kernelName, const dim3& numblocks, const dim3& numthreads, - unsigned memperblock, hipStream_t streamId, Args... args) { - kernelName<<>>(args...); -} +#define hipLaunchKernel(kernelName, numblocks, numthreads, memperblock, streamId, ...) \ + do { \ + kernelName<<<(numblocks), (numthreads), (memperblock), (streamId)>>>(hipLaunchParam{}, ##__VA_ARGS__); \ + } while (0) -template -inline void hipLaunchKernel(F&& kernel, const dim3& numBlocks, const dim3& dimBlocks, - std::uint32_t groupMemBytes, hipStream_t stream, Args... args) { - hipLaunchKernelGGL(kernel, numBlocks, dimBlocks, groupMemBytes, stream, hipLaunchParm{}, - std::move(args)...); -} +#define hipLaunchKernelGGL(kernelName, numblocks, numthreads, memperblock, streamId, ...) \ + do { \ + kernelName<<<(numblocks), (numthreads), (memperblock), (streamId)>>>(__VA_ARGS__); \ + } while (0) #include