| 59 | |
| 60 | template <typename Expected> |
| 61 | void AssertCudaBufferEquals(const CudaBuffer& buffer, Expected&& expected) { |
| 62 | ASSERT_OK_AND_ASSIGN(auto result, AllocateBuffer(buffer.size())); |
| 63 | ASSERT_OK(buffer.CopyToHost(0, buffer.size(), result->mutable_data())); |
| 64 | AssertBufferEqual(*result, expected); |
| 65 | } |
| 66 | |
| 67 | template <typename Expected> |
| 68 | void AssertCudaBufferEquals(const Buffer& buffer, Expected&& expected) { |
no test coverage detected