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

Method GetCudaStream

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

Source from the content-addressed store, hash-verified

54}
55
56CUstream NvOFCudaAPI::GetCudaStream(NV_OF_BUFFER_USAGE usage) {
57 CUstream stream = 0;
58 if (usage == NV_OF_BUFFER_USAGE_INPUT) {
59 stream = m_inputStream;
60 } else if (
61 (usage == NV_OF_BUFFER_USAGE_OUTPUT) ||
62 (usage == NV_OF_BUFFER_USAGE_COST) || (usage == NV_OF_BUFFER_USAGE_HINT)) {
63 stream = m_outputStream;
64 }
65 return stream;
66}
67
68NvOFObj NvOFCuda::Create(
69 CUcontext cuContext, uint32_t nWidth, uint32_t nHeight,

Callers 2

UploadDataMethod · 0.80
DownloadDataMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected