Function getDevice

src/caffe/parallel.cpp:110–114 · view source on GitHub ↗

Source from the content-addressed store, hash-verified

108	}
109
110	static int getDevice() {
111	int device = 0;
112	CUDA_CHECK(cudaGetDevice(&device));
113	return device;
114	}
115
116	template<typename Dtype>
117	NCCL<Dtype>::NCCL(shared_ptr<Solver<Dtype> > solver)

NCCLMethod · 0.85

no outgoing calls

no test coverage detected