Skip to content

Commit

Permalink
[CIR][CUDA][NFC] Skeleton of NVPTX target lowering info
Browse files Browse the repository at this point in the history
  • Loading branch information
AdUhTkJm committed Feb 19, 2025
1 parent 2df2022 commit 18e2cdb
Show file tree
Hide file tree
Showing 5 changed files with 82 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ add_clang_library(TargetLowering
TargetInfo.cpp
TargetLoweringInfo.cpp
Targets/AArch64.cpp
Targets/NVPTX.cpp
Targets/SPIR.cpp
Targets/X86.cpp
Targets/LoweringPrepareAArch64CXXABI.cpp
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -81,6 +81,8 @@ createTargetLoweringInfo(LowerModule &LM) {
}
case llvm::Triple::spirv64:
return createSPIRVTargetLoweringInfo(LM);
case llvm::Triple::nvptx64:
return createNVPTXTargetLoweringInfo(LM);
default:
cir_cconv_unreachable("ABI NYI");
}
Expand Down
3 changes: 3 additions & 0 deletions clang/lib/CIR/Dialect/Transforms/TargetLowering/TargetInfo.h
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,9 @@ createAArch64TargetLoweringInfo(LowerModule &CGM, cir::AArch64ABIKind AVXLevel);
std::unique_ptr<TargetLoweringInfo>
createSPIRVTargetLoweringInfo(LowerModule &CGM);

std::unique_ptr<TargetLoweringInfo>
createNVPTXTargetLoweringInfo(LowerModule &CGM);

} // namespace cir

#endif // LLVM_CLANG_LIB_CIR_DIALECT_TRANSFORMS_TARGETLOWERING_TARGETINFO_H
71 changes: 71 additions & 0 deletions clang/lib/CIR/Dialect/Transforms/TargetLowering/Targets/NVPTX.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,71 @@
//===- NVPTX.cpp - TargetInfo for NVPTX -----------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "ABIInfoImpl.h"
#include "LowerFunctionInfo.h"
#include "LowerTypes.h"
#include "TargetInfo.h"
#include "TargetLoweringInfo.h"
#include "clang/CIR/ABIArgInfo.h"
#include "clang/CIR/MissingFeatures.h"
#include "llvm/Support/ErrorHandling.h"

using ABIArgInfo = cir::ABIArgInfo;
using MissingFeature = cir::MissingFeatures;

namespace cir {

//===----------------------------------------------------------------------===//
// NVPTX ABI Implementation
//===----------------------------------------------------------------------===//

namespace {

class NVPTXABIInfo : public ABIInfo {
public:
NVPTXABIInfo(LowerTypes &lt) : ABIInfo(lt) {}

private:
void computeInfo(LowerFunctionInfo &fi) const override {
llvm_unreachable("NYI");
}
};

class NVPTXTargetLoweringInfo : public TargetLoweringInfo {
public:
NVPTXTargetLoweringInfo(LowerTypes &lt)
: TargetLoweringInfo(std::make_unique<NVPTXABIInfo>(lt)) {}

unsigned getTargetAddrSpaceFromCIRAddrSpace(
cir::AddressSpaceAttr addressSpaceAttr) const override {
using Kind = cir::AddressSpaceAttr::Kind;
switch (addressSpaceAttr.getValue()) {
case Kind::offload_private:
return 0;
case Kind::offload_local:
return 3;
case Kind::offload_global:
return 1;
case Kind::offload_constant:
return 2;
case Kind::offload_generic:
return 4;
default:
cir_cconv_unreachable("Unknown CIR address space for this target");
}
}
};

} // namespace

std::unique_ptr<TargetLoweringInfo>
createNVPTXTargetLoweringInfo(LowerModule &lowerModule) {
return std::make_unique<NVPTXTargetLoweringInfo>(lowerModule.getTypes());
}

} // namespace cir
5 changes: 5 additions & 0 deletions clang/lib/CIR/FrontendAction/CIRGenAction.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -260,6 +260,11 @@ class CIRGenConsumer : public clang::ASTConsumer {
}
}

if (action != CIRGenAction::OutputType::EmitCIR) {
if (C.getLangOpts().CUDA && !C.getLangOpts().CUDAIsDevice)
llvm_unreachable("CUDA NYI");
}

switch (action) {
case CIRGenAction::OutputType::EmitCIR:
case CIRGenAction::OutputType::EmitCIRFlat:
Expand Down

0 comments on commit 18e2cdb

Please sign in to comment.