Skip to content

Commit ac46df5

Browse files
authored
[CUDA][Integrate] Switch CUDATarget to ptx_kernel cc. (#19925)
This drops the local LLVM revert: llvm/llvm-project@de7438e Signed-off-by: hanhanW <[email protected]>
1 parent da4bdca commit ac46df5

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

compiler/plugins/target/CUDA/CUDATarget.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -545,7 +545,7 @@ class CUDATargetBackend final : public TargetBackend {
545545
};
546546

547547
// Mark the entry point as a kernel.
548-
setMetadataValueI32("kernel", 1);
548+
llvmFunc->setCallingConv(llvm::CallingConv::PTX_Kernel);
549549

550550
// Set the maximum number of threads in the thread block (CTA).
551551
auto exportOp = exportOpMap[funcOp.getName()];

0 commit comments

Comments
 (0)