MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / NvOFBufferCudaArray

Method NvOFBufferCudaArray

src/opr/impl/nvof/NvOFCuda.cpp:249–260  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

247}
248
249NvOFBufferCudaArray::NvOFBufferCudaArray(
250 std::shared_ptr<NvOFCudaAPI> ofAPI, const NV_OF_BUFFER_DESCRIPTOR& desc,
251 uint32_t elementSize)
252 : NvOFBuffer(desc, elementSize), m_cuArray(0), m_NvOFAPI(ofAPI) {
253 m_cuContext = m_NvOFAPI->GetCudaContext();
254 NVOF_API_CALL(m_NvOFAPI->GetAPI()->nvOFCreateGPUBufferCuda(
255 m_NvOFAPI->GetHandle(), &desc, NV_OF_CUDA_BUFFER_TYPE_CUARRAY,
256 &m_hGPUBuffer));
257 m_cuArray = m_NvOFAPI->GetAPI()->nvOFGPUBufferGetCUarray(m_hGPUBuffer);
258 NVOF_API_CALL(m_NvOFAPI->GetAPI()->nvOFGPUBufferGetStrideInfo(
259 m_hGPUBuffer, &m_strideInfo));
260}
261
262NvOFBufferCudaArray::~NvOFBufferCudaArray() {
263 m_NvOFAPI->GetAPI()->nvOFDestroyGPUBufferCuda(m_hGPUBuffer);

Callers

nothing calls this directly

Calls 3

GetCudaContextMethod · 0.80
GetAPIMethod · 0.80
GetHandleMethod · 0.80

Tested by

no test coverage detected