| /dports/misc/mnn/MNN-1.2.0/source/backend/cuda/core/runtime/ |
| H A D | CUDARuntime.cpp | 35 bool CUDARuntime::isCreateError() const { in isCreateError()
39 CUDARuntime::CUDARuntime(bool permitFloat16, int device_id) { in CUDARuntime() function in MNN::CUDARuntime
63 CUDARuntime::~CUDARuntime() { in ~CUDARuntime()
91 bool CUDARuntime::isSupportedFP16() const { in isSupportedFP16()
95 bool CUDARuntime::isSupportedDotInt8() const { in isSupportedDotInt8()
107 int CUDARuntime::device_id() const { in device_id()
111 void CUDARuntime::activate() { in activate()
118 void *CUDARuntime::alloc(size_t size_in_bytes) { in alloc()
125 void CUDARuntime::free(void *ptr) { in free()
152 cublasHandle_t CUDARuntime::cublas_handle() { in cublas_handle()
[all …]
|
| H A D | CUDARuntime.hpp | 83 class CUDARuntime { class
85 CUDARuntime(bool permitFloat16, int device_id);
86 ~CUDARuntime();
87 CUDARuntime(const CUDARuntime &) = delete;
88 CUDARuntime &operator=(const CUDARuntime &) = delete;
|
| /dports/misc/mnn/MNN-1.2.0/source/backend/cuda/execution/ |
| H A D | Raster.cuh | 7 …nt32_t* size, const int32_t* srcStride, const int32_t* dstStride, int bytes, CUDARuntime* runtime);
8 …8_t* dest, const uint8_t* src, int inside, int axis, int outside, int bytes, CUDARuntime* runtime);
9 …8_t* dest, const uint8_t* src, int inside, int axis, int outside, int bytes, CUDARuntime* runtime);
10 …srcStep, int srcLimit, const Tensor::InsideDescribe::Region& reg, int bytes, CUDARuntime* runtime);
11 …* size, const int32_t* srcStride, const int32_t* dstStride, int bytes, CUDARuntime* runtime, int o…
12 …nst int32_t* srcStride1, const int32_t* dstStride, halide_type_t type, CUDARuntime* runtime, int o…
|
| H A D | Raster.cu | 21 …output, const uint8_t* input, int inside, int axis, int outside, int bytes, CUDARuntime* runtime) { in PackC4()
57 …output, const uint8_t* input, int inside, int axis, int outside, int bytes, CUDARuntime* runtime) { in UnpackC4()
123 …rcStep, int srcLimit, const Tensor::InsideDescribe::Region& reg, int bytes, CUDARuntime* runtime) { in BlitWithIndice()
214 …t32_t* size, const int32_t* srcStride, const int32_t* dstStride, int bytes, CUDARuntime* runtime) { in RasterBlit()
242 …* size, const int32_t* srcStride, const int32_t* dstStride, int bytes, CUDARuntime* runtime, int o… in UnaryBlit()
337 …tride, const int32_t* srcStride1, const int32_t* dstStride, int bytes, CUDARuntime* runtime, int o… in BinaryBlitTemplateFloat()
374 …tride, const int32_t* srcStride1, const int32_t* dstStride, int bytes, CUDARuntime* runtime, int o… in BinaryBlitTemplateInt32()
406 …nst int32_t* srcStride1, const int32_t* dstStride, halide_type_t type, CUDARuntime* runtime, int o… in BinaryBlit()
|
| H A D | UnaryExecution.hpp | 29 CUDARuntime *mRuntime;
|
| H A D | ScatterNdExecution.hpp | 29 CUDARuntime *mRuntime;
|
| H A D | PReLUExecution.hpp | 29 CUDARuntime *mRuntime;
|
| H A D | InterpExecution.hpp | 29 CUDARuntime *mRuntime;
|
| H A D | ScaleExecution.hpp | 29 CUDARuntime *mRuntime;
|
| H A D | LayerNormExecution.hpp | 29 CUDARuntime *mRuntime;
|
| /dports/misc/mnn/MNN-1.2.0/source/backend/cuda/core/ |
| H A D | CUDABackend.hpp | 38 std::shared_ptr<CUDARuntime> mCUDARuntime;
44 CUDABackend(std::shared_ptr<BufferAllocator> st, std::shared_ptr<CUDARuntime> rt);
47 CUDARuntime *getCUDARuntime();
84 std::shared_ptr<CUDARuntime> mCUDARuntime;
|
| H A D | CUDABackend.cpp | 29 CUDARuntimeAllocator(CUDARuntime* rt) : mRuntime(rt) { in CUDARuntimeAllocator()
40 CUDARuntime* mRuntime;
45 mCUDARuntime.reset(new CUDARuntime(true, -1)); in CUDARuntimeWrapper()
47 mCUDARuntime.reset(new CUDARuntime(false, -1)); in CUDARuntimeWrapper()
75 std::shared_ptr<CUDARuntime> rt) in CUDABackend()
88 CUDARuntime* CUDABackend::getCUDARuntime() { in getCUDARuntime()
|
| /dports/devel/llvm70/llvm-7.0.1.src/tools/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 297 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
556 assert(CUDARuntime != nullptr); in getCUDARuntime()
557 return *CUDARuntime; in getCUDARuntime()
|
| /dports/devel/llvm-devel/llvm-project-f05c95f10fc1d8171071735af8ad3a9e87633120/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 331 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
623 assert(CUDARuntime != nullptr); in getCUDARuntime()
624 return *CUDARuntime; in getCUDARuntime()
|
| /dports/www/chromium-legacy/chromium-88.0.4324.182/third_party/llvm/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 327 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
600 assert(CUDARuntime != nullptr); in getCUDARuntime()
601 return *CUDARuntime; in getCUDARuntime()
|
| /dports/devel/wasi-libcxx/llvm-project-13.0.1.src/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 331 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
623 assert(CUDARuntime != nullptr); in getCUDARuntime()
624 return *CUDARuntime; in getCUDARuntime()
|
| /dports/devel/llvm12/llvm-project-12.0.1.src/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 327 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
600 assert(CUDARuntime != nullptr); in getCUDARuntime()
601 return *CUDARuntime; in getCUDARuntime()
|
| /dports/devel/llvm11/llvm-11.0.1.src/tools/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 327 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
601 assert(CUDARuntime != nullptr); in getCUDARuntime()
602 return *CUDARuntime; in getCUDARuntime()
|
| /dports/devel/llvm-cheri/llvm-project-37c49ff00e3eadce5d8703fdc4497f28458c64a8/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 327 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
601 assert(CUDARuntime != nullptr); in getCUDARuntime()
602 return *CUDARuntime; in getCUDARuntime()
|
| /dports/devel/llvm10/llvm-10.0.1.src/tools/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 326 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
597 assert(CUDARuntime != nullptr); in getCUDARuntime()
598 return *CUDARuntime; in getCUDARuntime()
|
| /dports/devel/wasi-compiler-rt13/llvm-project-13.0.1.src/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 331 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
623 assert(CUDARuntime != nullptr); in getCUDARuntime()
624 return *CUDARuntime; in getCUDARuntime()
|
| /dports/devel/llvm90/llvm-9.0.1.src/tools/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 322 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
590 assert(CUDARuntime != nullptr); in getCUDARuntime()
591 return *CUDARuntime; in getCUDARuntime()
|
| /dports/devel/tinygo/tinygo-0.14.1/llvm-project/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 326 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
597 assert(CUDARuntime != nullptr); in getCUDARuntime()
598 return *CUDARuntime; in getCUDARuntime()
|
| /dports/lang/rust/rustc-1.58.1-src/src/llvm-project/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 331 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
623 assert(CUDARuntime != nullptr); in getCUDARuntime()
624 return *CUDARuntime; in getCUDARuntime()
|
| /dports/devel/wasi-compiler-rt12/llvm-project-12.0.1.src/clang/lib/CodeGen/ |
| H A D | CodeGenModule.h | 327 std::unique_ptr<CGCUDARuntime> CUDARuntime; variable
600 assert(CUDARuntime != nullptr); in getCUDARuntime()
601 return *CUDARuntime; in getCUDARuntime()
|