Method set_device

src/core/test/mem_alloc.cpp:516–516

class CudaDevicePolicy : public DevicePolicy {
public:
    void set_device(int device) override { MGB_CUDA_CHECK(cudaSetDevice(device)); }
    void get_mem_info(size_t& free, size_t& tot) override {
        MGB_CUDA_CHECK(cudaMemGetInfo(&free, &tot));
    }