Searched refs:CUDA_REPORT_IF_ERROR (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/mlir/tools/mlir-cuda-runner/ |
| D | cuda-runtime-wrappers.cpp | 24 #define CUDA_REPORT_IF_ERROR(expr) \ macro
37 CUDA_REPORT_IF_ERROR(cuInit(/*flags=*/0)); in __anon73f9f5f60102()
39 CUDA_REPORT_IF_ERROR(cuDeviceGet(&device, /*ordinal=*/0)); in __anon73f9f5f60102()
41 CUDA_REPORT_IF_ERROR(cuCtxCreate(&context, /*flags=*/0, device)); in __anon73f9f5f60102()
47 CUDA_REPORT_IF_ERROR(cuModuleLoadData(&module, data)); in mgpuModuleLoad()
52 CUDA_REPORT_IF_ERROR(cuModuleUnload(module)); in mgpuModuleUnload()
57 CUDA_REPORT_IF_ERROR(cuModuleGetFunction(&function, module, name)); in mgpuModuleGetFunction()
69 CUDA_REPORT_IF_ERROR(cuLaunchKernel(function, gridX, gridY, gridZ, blockX, in mgpuLaunchKernel()
76 CUDA_REPORT_IF_ERROR(cuStreamCreate(&stream, CU_STREAM_NON_BLOCKING)); in mgpuStreamCreate()
81 CUDA_REPORT_IF_ERROR(cuStreamDestroy(stream)); in mgpuStreamDestroy()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_cuda_runtime_wrappers.cc | 48 #define CUDA_REPORT_IF_ERROR(expr) CUDA_REPORT_IF_ERROR_WITH_CTX(expr, nullptr) macro
59 CUDA_REPORT_IF_ERROR(cuModuleLoadData(&module, data)); in loadModule()
72 CUDA_REPORT_IF_ERROR(cuStreamGetCtx(stream, &context)); in get()
|