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

Method BenchmarkEnv

dnn/test/cuda/chanwise_convolution3d.cpp:48–58  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

46
47public:
48 BenchmarkEnv(Handle* handle, Handle* handle_cpu) {
49 this->handle = handle;
50 this->handle_cpu = handle_cpu;
51 rng = handle->create_operator<GaussianRNG>();
52 // make cpu handle used
53 handle_cpu->create_operator<Sleep>()->exec();
54
55 for (int i = 0; i < 3; ++i)
56 cudaEventCreate(&cuda_ev[i]);
57 megcoreGetCUDAStream(handle->megcore_computing_handle(), &cuda_stream);
58 }
59
60 ~BenchmarkEnv() {
61 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