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

Method BenchmarkEnv

dnn/test/cuda/chanwise_convolution.cpp:55–65  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

53
54public:
55 BenchmarkEnv(Handle* handle, Handle* handle_cpu) {
56 this->handle = handle;
57 this->handle_cpu = handle_cpu;
58 rng = handle->create_operator<GaussianRNG>();
59 // make cpu handle used
60 handle_cpu->create_operator<Sleep>()->exec();
61
62 for (int i = 0; i < 3; ++i)
63 cudaEventCreate(&cuda_ev[i]);
64 megcoreGetCUDAStream(handle->megcore_computing_handle(), &cuda_stream);
65 }
66
67 ~BenchmarkEnv() {
68 for (int i = 0; i < 3; ++i)

Callers

nothing calls this directly

Calls 3

megcoreGetCUDAStreamFunction · 0.85
execMethod · 0.45

Tested by

no test coverage detected