| 74 | class TestCudaBase : public ::testing::Test { |
| 75 | public: |
| 76 | void SetUp() { |
| 77 | ASSERT_OK_AND_ASSIGN(manager_, CudaDeviceManager::Instance()); |
| 78 | ASSERT_OK_AND_ASSIGN(device_, manager_->GetDevice(kGpuNumber)); |
| 79 | // ASSERT_OK(device_->GetContext(kGpuNumber, &context_)); |
| 80 | ASSERT_OK_AND_ASSIGN(context_, device_->GetContext()); |
| 81 | ASSERT_OK_AND_ASSIGN(mm_, AsCudaMemoryManager(device_->default_memory_manager())); |
| 82 | cpu_device_ = CPUDevice::Instance(); |
| 83 | cpu_mm_ = cpu_device_->default_memory_manager(); |
| 84 | } |
| 85 | |
| 86 | void TearDown() { |
| 87 | for (auto cu_context : non_primary_contexts_) { |
nothing calls this directly
no test coverage detected