mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2026-10-25 22:28:17 +08:00
Apply consistent code formatting across the repo. Add clang-format and pre-commit hooks.
This commit is contained in:
Rob Armstrong
@@ -24,6 +24,7 @@
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#include <cassert>
|
||||
#include <cuda.h>
|
||||
#include <llvm/ADT/StringExtras.h>
|
||||
#include <llvm/IR/IRBuilder.h>
|
||||
@@ -34,10 +35,8 @@
|
||||
#include <llvm/Support/Path.h>
|
||||
#include <llvm/Support/Program.h>
|
||||
#include <llvm/Support/raw_ostream.h>
|
||||
#include <nvvm.h>
|
||||
|
||||
#include <cassert>
|
||||
#include <memory>
|
||||
#include <nvvm.h>
|
||||
#include <string>
|
||||
|
||||
#include "DDSWriter.h"
|
||||
@@ -45,289 +44,282 @@
|
||||
static_assert(sizeof(void *) == 8, "Only 64bit targets are supported.");
|
||||
using namespace llvm;
|
||||
|
||||
static cl::opt<bool> SaveCubin("save-cubin",
|
||||
cl::desc("Write linker cubin to disk"),
|
||||
cl::init(false));
|
||||
static cl::opt<bool> SaveIR("save-ir", cl::desc("Write LLVM IR to disk"),
|
||||
cl::init(false));
|
||||
static cl::opt<bool> SavePTX("save-ptx", cl::desc("Write PTX to disk"),
|
||||
cl::init(false));
|
||||
static cl::opt<bool> SaveCubin("save-cubin", cl::desc("Write linker cubin to disk"), cl::init(false));
|
||||
static cl::opt<bool> SaveIR("save-ir", cl::desc("Write LLVM IR to disk"), cl::init(false));
|
||||
static cl::opt<bool> SavePTX("save-ptx", cl::desc("Write PTX to disk"), cl::init(false));
|
||||
|
||||
// Width and height of the output image.
|
||||
const unsigned width = 1024;
|
||||
const unsigned width = 1024;
|
||||
const unsigned height = 512;
|
||||
|
||||
// If 'err' is non-zero, emit an error message and exit.
|
||||
#define checkCudaErrors(err) __checkCudaErrors(err, __FILE__, __LINE__)
|
||||
static void __checkCudaErrors(CUresult err, const char *filename, int line) {
|
||||
assert(filename);
|
||||
if (CUDA_SUCCESS != err) {
|
||||
const char *ename = NULL;
|
||||
const CUresult res = cuGetErrorName(err, &ename);
|
||||
fprintf(stderr,
|
||||
"CUDA API Error %04d: \"%s\" from file <%s>, "
|
||||
"line %i.\n",
|
||||
err, ((CUDA_SUCCESS == res) ? ename : "Unknown"), filename, line);
|
||||
exit(err);
|
||||
}
|
||||
static void __checkCudaErrors(CUresult err, const char *filename, int line)
|
||||
{
|
||||
assert(filename);
|
||||
if (CUDA_SUCCESS != err) {
|
||||
const char *ename = NULL;
|
||||
const CUresult res = cuGetErrorName(err, &ename);
|
||||
fprintf(stderr,
|
||||
"CUDA API Error %04d: \"%s\" from file <%s>, "
|
||||
"line %i.\n",
|
||||
err,
|
||||
((CUDA_SUCCESS == res) ? ename : "Unknown"),
|
||||
filename,
|
||||
line);
|
||||
exit(err);
|
||||
}
|
||||
}
|
||||
|
||||
// Verify that the NVVM result code is success, or terminate otherwise.
|
||||
void checkNVVMCall(nvvmResult res) {
|
||||
if (res != NVVM_SUCCESS) {
|
||||
errs() << "libnvvm call failed\n";
|
||||
exit(res);
|
||||
}
|
||||
void checkNVVMCall(nvvmResult res)
|
||||
{
|
||||
if (res != NVVM_SUCCESS) {
|
||||
errs() << "libnvvm call failed\n";
|
||||
exit(res);
|
||||
}
|
||||
}
|
||||
|
||||
/// generateModule - Generate and LLVM IR module that calls an
|
||||
/// externally-defined function
|
||||
std::unique_ptr<Module> generateModule(LLVMContext &context) {
|
||||
// Create the module and setup the layout and triple.
|
||||
auto mod = std::make_unique<Module>("nvvm-module", context);
|
||||
mod->setDataLayout(
|
||||
"e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-i128:128:128-"
|
||||
"f32:32:32-f64:64:64-v16:16:16-v32:32:32-v64:64:64-v128:128:128-n16:32:"
|
||||
"64");
|
||||
mod->setTargetTriple("nvptx64-nvidia-cuda");
|
||||
std::unique_ptr<Module> generateModule(LLVMContext &context)
|
||||
{
|
||||
// Create the module and setup the layout and triple.
|
||||
auto mod = std::make_unique<Module>("nvvm-module", context);
|
||||
mod->setDataLayout("e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-i128:128:128-"
|
||||
"f32:32:32-f64:64:64-v16:16:16-v32:32:32-v64:64:64-v128:128:128-n16:32:"
|
||||
"64");
|
||||
mod->setTargetTriple("nvptx64-nvidia-cuda");
|
||||
|
||||
// Get pointers to some commonly-used types.
|
||||
Type *voidTy = Type::getVoidTy(context);
|
||||
Type *floatTy = Type::getFloatTy(context);
|
||||
Type *i32Ty = Type::getInt32Ty(context);
|
||||
Type *floatGenericPtrTy = PointerType::get(floatTy, /* address space */ 0);
|
||||
// Get pointers to some commonly-used types.
|
||||
Type *voidTy = Type::getVoidTy(context);
|
||||
Type *floatTy = Type::getFloatTy(context);
|
||||
Type *i32Ty = Type::getInt32Ty(context);
|
||||
Type *floatGenericPtrTy = PointerType::get(floatTy, /* address space */ 0);
|
||||
|
||||
// void @mandelbrot(float*)
|
||||
Type *mandelbrotParamTys[] = {floatGenericPtrTy};
|
||||
FunctionType *mandelbrotTy =
|
||||
FunctionType::get(voidTy, mandelbrotParamTys, false);
|
||||
FunctionCallee mandelbrotFunc =
|
||||
mod->getOrInsertFunction("mandelbrot", mandelbrotTy);
|
||||
// void @mandelbrot(float*)
|
||||
Type *mandelbrotParamTys[] = {floatGenericPtrTy};
|
||||
FunctionType *mandelbrotTy = FunctionType::get(voidTy, mandelbrotParamTys, false);
|
||||
FunctionCallee mandelbrotFunc = mod->getOrInsertFunction("mandelbrot", mandelbrotTy);
|
||||
|
||||
// Kernel argument types.
|
||||
Type *paramTys[] = {floatGenericPtrTy};
|
||||
// Kernel argument types.
|
||||
Type *paramTys[] = {floatGenericPtrTy};
|
||||
|
||||
// Kernel function type.
|
||||
FunctionType *funcTy = FunctionType::get(voidTy, paramTys, false);
|
||||
// Kernel function type.
|
||||
FunctionType *funcTy = FunctionType::get(voidTy, paramTys, false);
|
||||
|
||||
// Kernel function.
|
||||
Function *func =
|
||||
Function::Create(funcTy, GlobalValue::ExternalLinkage, "kernel", *mod);
|
||||
func->arg_begin()->setName("ptr");
|
||||
// Kernel function.
|
||||
Function *func = Function::Create(funcTy, GlobalValue::ExternalLinkage, "kernel", *mod);
|
||||
func->arg_begin()->setName("ptr");
|
||||
|
||||
// 'entry' basic block in kernel function.
|
||||
BasicBlock *entry = BasicBlock::Create(context, "entry", func);
|
||||
// 'entry' basic block in kernel function.
|
||||
BasicBlock *entry = BasicBlock::Create(context, "entry", func);
|
||||
|
||||
// Build the entry block.
|
||||
IRBuilder<> builder(entry);
|
||||
builder.CreateCall(mandelbrotFunc, func->arg_begin());
|
||||
builder.CreateRetVoid();
|
||||
// Build the entry block.
|
||||
IRBuilder<> builder(entry);
|
||||
builder.CreateCall(mandelbrotFunc, func->arg_begin());
|
||||
builder.CreateRetVoid();
|
||||
|
||||
// Create kernel metadata.
|
||||
Metadata *mdVals[] = {ValueAsMetadata::get(func),
|
||||
MDString::get(context, "kernel"),
|
||||
ConstantAsMetadata::get(ConstantInt::getTrue(context))};
|
||||
MDNode *kernelMD = MDNode::get(context, mdVals);
|
||||
NamedMDNode *nvvmAnnot = mod->getOrInsertNamedMetadata("nvvm.annotations");
|
||||
nvvmAnnot->addOperand(kernelMD);
|
||||
// Create kernel metadata.
|
||||
Metadata *mdVals[] = {ValueAsMetadata::get(func),
|
||||
MDString::get(context, "kernel"),
|
||||
ConstantAsMetadata::get(ConstantInt::getTrue(context))};
|
||||
MDNode *kernelMD = MDNode::get(context, mdVals);
|
||||
NamedMDNode *nvvmAnnot = mod->getOrInsertNamedMetadata("nvvm.annotations");
|
||||
nvvmAnnot->addOperand(kernelMD);
|
||||
|
||||
// Set the NVVM IR version to 2.0.
|
||||
auto *two = ConstantInt::get(Type::getInt32Ty(context), 2);
|
||||
auto *zero = ConstantInt::get(Type::getInt32Ty(context), 0);
|
||||
auto *versionMD = MDNode::get(
|
||||
context, {ConstantAsMetadata::get(two), ConstantAsMetadata::get(zero)});
|
||||
NamedMDNode *nvvmIRVersion = mod->getOrInsertNamedMetadata("nvvmir.version");
|
||||
nvvmIRVersion->addOperand(versionMD);
|
||||
// Set the NVVM IR version to 2.0.
|
||||
auto *two = ConstantInt::get(Type::getInt32Ty(context), 2);
|
||||
auto *zero = ConstantInt::get(Type::getInt32Ty(context), 0);
|
||||
auto *versionMD = MDNode::get(context, {ConstantAsMetadata::get(two), ConstantAsMetadata::get(zero)});
|
||||
NamedMDNode *nvvmIRVersion = mod->getOrInsertNamedMetadata("nvvmir.version");
|
||||
nvvmIRVersion->addOperand(versionMD);
|
||||
|
||||
return mod;
|
||||
return mod;
|
||||
}
|
||||
|
||||
// Use libNVVM to compile an NVVM IR module to PTX.
|
||||
std::string generatePtx(const std::string &module, int devMajor, int devMinor,
|
||||
const char *moduleName) {
|
||||
assert(moduleName);
|
||||
std::string generatePtx(const std::string &module, int devMajor, int devMinor, const char *moduleName)
|
||||
{
|
||||
assert(moduleName);
|
||||
|
||||
// libNVVM initialization.
|
||||
nvvmProgram compileUnit;
|
||||
checkNVVMCall(nvvmCreateProgram(&compileUnit));
|
||||
// libNVVM initialization.
|
||||
nvvmProgram compileUnit;
|
||||
checkNVVMCall(nvvmCreateProgram(&compileUnit));
|
||||
|
||||
// Create a libNVVM compilation unit from the NVVM IR.
|
||||
checkNVVMCall(nvvmAddModuleToProgram(compileUnit, module.c_str(),
|
||||
module.size(), moduleName));
|
||||
std::string computeArg = "-arch=compute_";
|
||||
computeArg += utostr(devMajor);
|
||||
computeArg += utostr(devMinor);
|
||||
// Create a libNVVM compilation unit from the NVVM IR.
|
||||
checkNVVMCall(nvvmAddModuleToProgram(compileUnit, module.c_str(), module.size(), moduleName));
|
||||
std::string computeArg = "-arch=compute_";
|
||||
computeArg += utostr(devMajor);
|
||||
computeArg += utostr(devMinor);
|
||||
|
||||
// Compile the NVVM IR into PTX.
|
||||
const char *options[] = {computeArg.c_str()};
|
||||
nvvmResult res = nvvmCompileProgram(compileUnit, 1, options);
|
||||
if (res != NVVM_SUCCESS) {
|
||||
errs() << "nvvmCompileProgram failed!\n";
|
||||
size_t logSize;
|
||||
nvvmGetProgramLogSize(compileUnit, &logSize);
|
||||
char *msg = new char[logSize];
|
||||
nvvmGetProgramLog(compileUnit, msg);
|
||||
errs() << msg << "\n";
|
||||
delete[] msg;
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
// Compile the NVVM IR into PTX.
|
||||
const char *options[] = {computeArg.c_str()};
|
||||
nvvmResult res = nvvmCompileProgram(compileUnit, 1, options);
|
||||
if (res != NVVM_SUCCESS) {
|
||||
errs() << "nvvmCompileProgram failed!\n";
|
||||
size_t logSize;
|
||||
nvvmGetProgramLogSize(compileUnit, &logSize);
|
||||
char *msg = new char[logSize];
|
||||
nvvmGetProgramLog(compileUnit, msg);
|
||||
errs() << msg << "\n";
|
||||
delete[] msg;
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
// Get the result PTX size and source.
|
||||
size_t ptxSize = 0;
|
||||
checkNVVMCall(nvvmGetCompiledResultSize(compileUnit, &ptxSize));
|
||||
char *ptx = new char[ptxSize];
|
||||
checkNVVMCall(nvvmGetCompiledResult(compileUnit, ptx));
|
||||
// Get the result PTX size and source.
|
||||
size_t ptxSize = 0;
|
||||
checkNVVMCall(nvvmGetCompiledResultSize(compileUnit, &ptxSize));
|
||||
char *ptx = new char[ptxSize];
|
||||
checkNVVMCall(nvvmGetCompiledResult(compileUnit, ptx));
|
||||
|
||||
// Clean-up libNVVM.
|
||||
checkNVVMCall(nvvmDestroyProgram(&compileUnit));
|
||||
// Clean-up libNVVM.
|
||||
checkNVVMCall(nvvmDestroyProgram(&compileUnit));
|
||||
|
||||
return std::string(ptx);
|
||||
return std::string(ptx);
|
||||
}
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
cl::ParseCommandLineOptions(argc, argv, "cuda-c-linking");
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
cl::ParseCommandLineOptions(argc, argv, "cuda-c-linking");
|
||||
|
||||
// Locate the pre-built library.
|
||||
std::string libpath0 = sys::fs::getMainExecutable(argv[0], (void *)main);
|
||||
SmallString<256> libpath(libpath0);
|
||||
const char *mathlibFile = "libmathfuncs64.a";
|
||||
sys::path::remove_filename(libpath);
|
||||
sys::path::append(libpath, mathlibFile);
|
||||
// Locate the pre-built library.
|
||||
std::string libpath0 = sys::fs::getMainExecutable(argv[0], (void *)main);
|
||||
SmallString<256> libpath(libpath0);
|
||||
const char *mathlibFile = "libmathfuncs64.a";
|
||||
sys::path::remove_filename(libpath);
|
||||
sys::path::append(libpath, mathlibFile);
|
||||
|
||||
if (!sys::fs::exists(libpath.c_str())) {
|
||||
errs() << "Unable to locate math library, expected at " << libpath << '\n';
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
if (!sys::fs::exists(libpath.c_str())) {
|
||||
errs() << "Unable to locate math library, expected at " << libpath << '\n';
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
outs() << "Using math library: " << libpath.str() << "\n";
|
||||
outs() << "Using math library: " << libpath.str() << "\n";
|
||||
|
||||
// Initialize CUDA and obtain device 0.
|
||||
checkCudaErrors(cuInit(0));
|
||||
int nDevices;
|
||||
checkCudaErrors(cuDeviceGetCount(&nDevices));
|
||||
if (nDevices == 0) {
|
||||
errs() << "Failed to locate any CUDA compute devices.\n";
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
CUdevice device;
|
||||
checkCudaErrors(cuDeviceGet(&device, 0));
|
||||
// Initialize CUDA and obtain device 0.
|
||||
checkCudaErrors(cuInit(0));
|
||||
int nDevices;
|
||||
checkCudaErrors(cuDeviceGetCount(&nDevices));
|
||||
if (nDevices == 0) {
|
||||
errs() << "Failed to locate any CUDA compute devices.\n";
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
CUdevice device;
|
||||
checkCudaErrors(cuDeviceGet(&device, 0));
|
||||
|
||||
char name[128];
|
||||
checkCudaErrors(cuDeviceGetName(name, 128, device));
|
||||
outs() << "Using CUDA Device [0]: " << name << "\n";
|
||||
char name[128];
|
||||
checkCudaErrors(cuDeviceGetName(name, 128, device));
|
||||
outs() << "Using CUDA Device [0]: " << name << "\n";
|
||||
|
||||
int devMajor = 0, devMinor = 0;
|
||||
checkCudaErrors(cuDeviceGetAttribute(
|
||||
&devMajor, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR, device));
|
||||
checkCudaErrors(cuDeviceGetAttribute(
|
||||
&devMinor, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR, device));
|
||||
outs() << "Device Compute Capability: " << devMajor << "." << devMinor
|
||||
<< "\n";
|
||||
if (devMajor < 5) {
|
||||
errs() << "ERROR: Device 0 is not sm_50 or later.\n";
|
||||
return 1;
|
||||
}
|
||||
int devMajor = 0, devMinor = 0;
|
||||
checkCudaErrors(cuDeviceGetAttribute(&devMajor, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR, device));
|
||||
checkCudaErrors(cuDeviceGetAttribute(&devMinor, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR, device));
|
||||
outs() << "Device Compute Capability: " << devMajor << "." << devMinor << "\n";
|
||||
if (devMajor < 5) {
|
||||
errs() << "ERROR: Device 0 is not sm_50 or later.\n";
|
||||
return 1;
|
||||
}
|
||||
|
||||
// Generate the IR module
|
||||
LLVMContext ctx;
|
||||
std::string moduleStr;
|
||||
auto module = generateModule(ctx);
|
||||
// Generate the IR module
|
||||
LLVMContext ctx;
|
||||
std::string moduleStr;
|
||||
auto module = generateModule(ctx);
|
||||
|
||||
if (SaveIR) {
|
||||
std::error_code err;
|
||||
raw_fd_ostream out("cuda-c-linking.kernel.ll", err);
|
||||
out << *(module.get());
|
||||
}
|
||||
if (SaveIR) {
|
||||
std::error_code err;
|
||||
raw_fd_ostream out("cuda-c-linking.kernel.ll", err);
|
||||
out << *(module.get());
|
||||
}
|
||||
|
||||
// Write the module to a string.
|
||||
{
|
||||
llvm::raw_string_ostream str(moduleStr);
|
||||
str << *module.get();
|
||||
}
|
||||
// Write the module to a string.
|
||||
{
|
||||
llvm::raw_string_ostream str(moduleStr);
|
||||
str << *module.get();
|
||||
}
|
||||
|
||||
// Generate PTX.
|
||||
std::string ptx = generatePtx(moduleStr, devMajor, devMinor,
|
||||
module->getModuleIdentifier().c_str());
|
||||
if (SavePTX) {
|
||||
std::error_code err;
|
||||
raw_fd_ostream out("cuda-c-linking.kernel.ptx", err);
|
||||
out << ptx;
|
||||
}
|
||||
// Generate PTX.
|
||||
std::string ptx = generatePtx(moduleStr, devMajor, devMinor, module->getModuleIdentifier().c_str());
|
||||
if (SavePTX) {
|
||||
std::error_code err;
|
||||
raw_fd_ostream out("cuda-c-linking.kernel.ptx", err);
|
||||
out << ptx;
|
||||
}
|
||||
|
||||
// Create the CUDA context.
|
||||
CUcontext context;
|
||||
checkCudaErrors(cuCtxCreate(&context, 0, device));
|
||||
// Create the CUDA context.
|
||||
CUcontext context;
|
||||
checkCudaErrors(cuCtxCreate(&context, 0, device));
|
||||
|
||||
// Create a JIT linker and generate the result CUBIN.
|
||||
CUlinkState linker;
|
||||
char linkerInfo[1024]{};
|
||||
char linkerErrors[1024]{};
|
||||
CUjit_option linkerOptions[] = {
|
||||
CU_JIT_INFO_LOG_BUFFER, CU_JIT_INFO_LOG_BUFFER_SIZE_BYTES,
|
||||
CU_JIT_ERROR_LOG_BUFFER, CU_JIT_ERROR_LOG_BUFFER_SIZE_BYTES,
|
||||
CU_JIT_LOG_VERBOSE};
|
||||
void *linkerOptionValues[] = {linkerInfo, reinterpret_cast<void *>(1024),
|
||||
linkerErrors, reinterpret_cast<void *>(1024),
|
||||
reinterpret_cast<void *>(1)};
|
||||
checkCudaErrors(cuLinkCreate(5, linkerOptions, linkerOptionValues, &linker));
|
||||
checkCudaErrors(cuLinkAddData(linker, CU_JIT_INPUT_PTX, (void *)ptx.c_str(),
|
||||
ptx.size(), "<compiled-ptx>", 0, NULL, NULL));
|
||||
checkCudaErrors(cuLinkAddFile(linker, CU_JIT_INPUT_LIBRARY, libpath.c_str(),
|
||||
0, NULL, NULL));
|
||||
void *cubin;
|
||||
size_t cubinSize;
|
||||
checkCudaErrors(cuLinkComplete(linker, &cubin, &cubinSize));
|
||||
outs() << "Linker Log:\n" << linkerInfo << "\n" << linkerErrors << "\n";
|
||||
if (SaveCubin) {
|
||||
std::error_code err;
|
||||
raw_fd_ostream out("cuda-c-linking.linked.cubin", err, sys::fs::OF_None);
|
||||
out.write(reinterpret_cast<char *>(cubin), cubinSize);
|
||||
}
|
||||
// Create a JIT linker and generate the result CUBIN.
|
||||
CUlinkState linker;
|
||||
char linkerInfo[1024]{};
|
||||
char linkerErrors[1024]{};
|
||||
CUjit_option linkerOptions[] = {CU_JIT_INFO_LOG_BUFFER,
|
||||
CU_JIT_INFO_LOG_BUFFER_SIZE_BYTES,
|
||||
CU_JIT_ERROR_LOG_BUFFER,
|
||||
CU_JIT_ERROR_LOG_BUFFER_SIZE_BYTES,
|
||||
CU_JIT_LOG_VERBOSE};
|
||||
void *linkerOptionValues[] = {linkerInfo,
|
||||
reinterpret_cast<void *>(1024),
|
||||
linkerErrors,
|
||||
reinterpret_cast<void *>(1024),
|
||||
reinterpret_cast<void *>(1)};
|
||||
checkCudaErrors(cuLinkCreate(5, linkerOptions, linkerOptionValues, &linker));
|
||||
checkCudaErrors(
|
||||
cuLinkAddData(linker, CU_JIT_INPUT_PTX, (void *)ptx.c_str(), ptx.size(), "<compiled-ptx>", 0, NULL, NULL));
|
||||
checkCudaErrors(cuLinkAddFile(linker, CU_JIT_INPUT_LIBRARY, libpath.c_str(), 0, NULL, NULL));
|
||||
void *cubin;
|
||||
size_t cubinSize;
|
||||
checkCudaErrors(cuLinkComplete(linker, &cubin, &cubinSize));
|
||||
outs() << "Linker Log:\n" << linkerInfo << "\n" << linkerErrors << "\n";
|
||||
if (SaveCubin) {
|
||||
std::error_code err;
|
||||
raw_fd_ostream out("cuda-c-linking.linked.cubin", err, sys::fs::OF_None);
|
||||
out.write(reinterpret_cast<char *>(cubin), cubinSize);
|
||||
}
|
||||
|
||||
// Create a module and load the cubin into it.
|
||||
CUmodule cudaModule;
|
||||
checkCudaErrors(cuModuleLoadDataEx(&cudaModule, cubin, 0, 0, 0));
|
||||
// Create a module and load the cubin into it.
|
||||
CUmodule cudaModule;
|
||||
checkCudaErrors(cuModuleLoadDataEx(&cudaModule, cubin, 0, 0, 0));
|
||||
|
||||
// Now that the CUBIN is loaded, we can release the linker.
|
||||
checkCudaErrors(cuLinkDestroy(linker));
|
||||
// Now that the CUBIN is loaded, we can release the linker.
|
||||
checkCudaErrors(cuLinkDestroy(linker));
|
||||
|
||||
// Get kernel function.
|
||||
CUfunction function;
|
||||
checkCudaErrors(cuModuleGetFunction(&function, cudaModule, "kernel"));
|
||||
// Get kernel function.
|
||||
CUfunction function;
|
||||
checkCudaErrors(cuModuleGetFunction(&function, cudaModule, "kernel"));
|
||||
|
||||
// Device data.
|
||||
CUdeviceptr devBuffer;
|
||||
checkCudaErrors(cuMemAlloc(&devBuffer, sizeof(float) * width * height * 4));
|
||||
float *data = new float[width * height * 4];
|
||||
// Device data.
|
||||
CUdeviceptr devBuffer;
|
||||
checkCudaErrors(cuMemAlloc(&devBuffer, sizeof(float) * width * height * 4));
|
||||
float *data = new float[width * height * 4];
|
||||
|
||||
// Each thread will generate one pixel, and we'll subdivide the problem into
|
||||
// 16x16 chunks.
|
||||
const unsigned blockSizeX = 16;
|
||||
const unsigned blockSizeY = 16;
|
||||
const unsigned blockSizeZ = 1;
|
||||
const unsigned gridSizeX = width / 16;
|
||||
const unsigned gridSizeY = height / 16;
|
||||
const unsigned gridSizeZ = 1;
|
||||
// Each thread will generate one pixel, and we'll subdivide the problem into
|
||||
// 16x16 chunks.
|
||||
const unsigned blockSizeX = 16;
|
||||
const unsigned blockSizeY = 16;
|
||||
const unsigned blockSizeZ = 1;
|
||||
const unsigned gridSizeX = width / 16;
|
||||
const unsigned gridSizeY = height / 16;
|
||||
const unsigned gridSizeZ = 1;
|
||||
|
||||
// Execute the kernel.
|
||||
outs() << "Launching kernel\n";
|
||||
void *params[] = {&devBuffer};
|
||||
checkCudaErrors(cuLaunchKernel(function, gridSizeX, gridSizeY, gridSizeZ,
|
||||
blockSizeX, blockSizeY, blockSizeZ, 0, NULL,
|
||||
params, NULL));
|
||||
// Execute the kernel.
|
||||
outs() << "Launching kernel\n";
|
||||
void *params[] = {&devBuffer};
|
||||
checkCudaErrors(cuLaunchKernel(
|
||||
function, gridSizeX, gridSizeY, gridSizeZ, blockSizeX, blockSizeY, blockSizeZ, 0, NULL, params, NULL));
|
||||
|
||||
// Retrieve the result data from the device.
|
||||
checkCudaErrors(
|
||||
cuMemcpyDtoH(&data[0], devBuffer, sizeof(float) * width * height * 4));
|
||||
// Retrieve the result data from the device.
|
||||
checkCudaErrors(cuMemcpyDtoH(&data[0], devBuffer, sizeof(float) * width * height * 4));
|
||||
|
||||
writeDDS("mandelbrot.dds", data, width, height);
|
||||
outs() << "Output saved to mandelbrot.dds\n";
|
||||
writeDDS("mandelbrot.dds", data, width, height);
|
||||
outs() << "Output saved to mandelbrot.dds\n";
|
||||
|
||||
// Cleanup.
|
||||
delete[] data;
|
||||
checkCudaErrors(cuMemFree(devBuffer));
|
||||
checkCudaErrors(cuModuleUnload(cudaModule));
|
||||
checkCudaErrors(cuCtxDestroy(context));
|
||||
// Cleanup.
|
||||
delete[] data;
|
||||
checkCudaErrors(cuMemFree(devBuffer));
|
||||
checkCudaErrors(cuModuleUnload(cudaModule));
|
||||
checkCudaErrors(cuCtxDestroy(context));
|
||||
|
||||
return 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
@@ -29,57 +29,56 @@
|
||||
|
||||
// Note that this kernel is meant to be a simple, straight-forward
|
||||
// implementation, and so may not represent optimized GPU code.
|
||||
extern "C"
|
||||
__device__
|
||||
void mandelbrot(float* Data) {
|
||||
extern "C" __device__ void mandelbrot(float *Data)
|
||||
{
|
||||
|
||||
// Which pixel am I?
|
||||
unsigned DataX = blockIdx.x * blockDim.x + threadIdx.x;
|
||||
unsigned DataY = blockIdx.y * blockDim.y + threadIdx.y;
|
||||
unsigned Width = gridDim.x * blockDim.x;
|
||||
unsigned Height = gridDim.y * blockDim.y;
|
||||
// Which pixel am I?
|
||||
unsigned DataX = blockIdx.x * blockDim.x + threadIdx.x;
|
||||
unsigned DataY = blockIdx.y * blockDim.y + threadIdx.y;
|
||||
unsigned Width = gridDim.x * blockDim.x;
|
||||
unsigned Height = gridDim.y * blockDim.y;
|
||||
|
||||
float R, G, B, A;
|
||||
float R, G, B, A;
|
||||
|
||||
// Scale coordinates to (-2.5, 1) and (-1, 1)
|
||||
// Scale coordinates to (-2.5, 1) and (-1, 1)
|
||||
|
||||
float NormX = (float)DataX / (float)Width;
|
||||
NormX *= 3.5f;
|
||||
NormX -= 2.5f;
|
||||
float NormX = (float)DataX / (float)Width;
|
||||
NormX *= 3.5f;
|
||||
NormX -= 2.5f;
|
||||
|
||||
float NormY = (float)DataY / (float)Height;
|
||||
NormY *= 2.0f;
|
||||
NormY -= 1.0f;
|
||||
float NormY = (float)DataY / (float)Height;
|
||||
NormY *= 2.0f;
|
||||
NormY -= 1.0f;
|
||||
|
||||
float X0 = NormX;
|
||||
float Y0 = NormY;
|
||||
float X0 = NormX;
|
||||
float Y0 = NormY;
|
||||
|
||||
float X = 0.0f;
|
||||
float Y = 0.0f;
|
||||
float X = 0.0f;
|
||||
float Y = 0.0f;
|
||||
|
||||
unsigned Iter = 0;
|
||||
unsigned MaxIter = 1000;
|
||||
unsigned Iter = 0;
|
||||
unsigned MaxIter = 1000;
|
||||
|
||||
// Iterate
|
||||
while(X*X + Y*Y < 4.0f && Iter < MaxIter) {
|
||||
float XTemp = X*X - Y*Y + X0;
|
||||
Y = 2.0f*X*Y + Y0;
|
||||
// Iterate
|
||||
while (X * X + Y * Y < 4.0f && Iter < MaxIter) {
|
||||
float XTemp = X * X - Y * Y + X0;
|
||||
Y = 2.0f * X * Y + Y0;
|
||||
|
||||
X = XTemp;
|
||||
X = XTemp;
|
||||
|
||||
Iter++;
|
||||
}
|
||||
Iter++;
|
||||
}
|
||||
|
||||
unsigned ColorG = Iter % 50;
|
||||
unsigned ColorB = Iter % 25;
|
||||
unsigned ColorG = Iter % 50;
|
||||
unsigned ColorB = Iter % 25;
|
||||
|
||||
R = 0.0f;
|
||||
G = (float)ColorG / 50.0f;
|
||||
B = (float)ColorB / 25.0f;
|
||||
A = 1.0f;
|
||||
R = 0.0f;
|
||||
G = (float)ColorG / 50.0f;
|
||||
B = (float)ColorB / 25.0f;
|
||||
A = 1.0f;
|
||||
|
||||
Data[DataY*Width*4+DataX*4+0] = R;
|
||||
Data[DataY*Width*4+DataX*4+1] = G;
|
||||
Data[DataY*Width*4+DataX*4+2] = B;
|
||||
Data[DataY*Width*4+DataX*4+3] = A;
|
||||
Data[DataY * Width * 4 + DataX * 4 + 0] = R;
|
||||
Data[DataY * Width * 4 + DataX * 4 + 1] = G;
|
||||
Data[DataY * Width * 4 + DataX * 4 + 2] = B;
|
||||
Data[DataY * Width * 4 + DataX * 4 + 3] = A;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user