3fef2d26a3
test/lib/Transforms/ has bitrot and become somewhat of a dumping grounds for testing pretty much any part of the project. This revision cleans this up, and moves the files within to a directory that reflects what is actually being tested. Differential Revision: https://reviews.llvm.org/D102456
71 lines
2.2 KiB
C++
71 lines
2.2 KiB
C++
//===- TestConvertGPUKernelToCubin.cpp - Test gpu kernel cubin lowering ---===//
|
|
//
|
|
// 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 "mlir/Dialect/GPU/Passes.h"
|
|
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Target/LLVMIR/Dialect/NVVM/NVVMToLLVMIRTranslation.h"
|
|
#include "mlir/Target/LLVMIR/Export.h"
|
|
#include "llvm/Support/TargetSelect.h"
|
|
|
|
using namespace mlir;
|
|
|
|
#if MLIR_CUDA_CONVERSIONS_ENABLED
|
|
namespace {
|
|
class TestSerializeToCubinPass
|
|
: public PassWrapper<TestSerializeToCubinPass, gpu::SerializeToBlobPass> {
|
|
public:
|
|
TestSerializeToCubinPass();
|
|
|
|
private:
|
|
void getDependentDialects(DialectRegistry ®istry) const override;
|
|
|
|
// Serializes PTX to CUBIN.
|
|
std::unique_ptr<std::vector<char>>
|
|
serializeISA(const std::string &isa) override;
|
|
};
|
|
} // namespace
|
|
|
|
TestSerializeToCubinPass::TestSerializeToCubinPass() {
|
|
this->triple = "nvptx64-nvidia-cuda";
|
|
this->chip = "sm_35";
|
|
this->features = "+ptx60";
|
|
}
|
|
|
|
void TestSerializeToCubinPass::getDependentDialects(
|
|
DialectRegistry ®istry) const {
|
|
registerNVVMDialectTranslation(registry);
|
|
gpu::SerializeToBlobPass::getDependentDialects(registry);
|
|
}
|
|
|
|
std::unique_ptr<std::vector<char>>
|
|
TestSerializeToCubinPass::serializeISA(const std::string &) {
|
|
std::string data = "CUBIN";
|
|
return std::make_unique<std::vector<char>>(data.begin(), data.end());
|
|
}
|
|
|
|
namespace mlir {
|
|
namespace test {
|
|
// Register test pass to serialize GPU module to a CUBIN binary annotation.
|
|
void registerTestGpuSerializeToCubinPass() {
|
|
PassRegistration<TestSerializeToCubinPass> registerSerializeToCubin(
|
|
"test-gpu-to-cubin",
|
|
"Lower GPU kernel function to CUBIN binary annotations", [] {
|
|
// Initialize LLVM NVPTX backend.
|
|
LLVMInitializeNVPTXTarget();
|
|
LLVMInitializeNVPTXTargetInfo();
|
|
LLVMInitializeNVPTXTargetMC();
|
|
LLVMInitializeNVPTXAsmPrinter();
|
|
|
|
return std::make_unique<TestSerializeToCubinPass>();
|
|
});
|
|
}
|
|
} // namespace test
|
|
} // namespace mlir
|
|
#endif // MLIR_CUDA_CONVERSIONS_ENABLED
|