def test_device_record_copy(self):
hostrec = self.hostz.copy()
devrec = DeviceRecord(self.dtype)
devrec.copy_to_device(hostrec)
# Copy back and check values are all zeros
hostrec2 = self.hostnz.copy()
devrec.copy_to_host(hostrec2)
np.testing.assert_equal(self.hostz, hostrec2)
# Copy non-zero values to GPU and back and check values
hostrec3 = self.hostnz.copy()
devrec.copy_to_device(hostrec3)
hostrec4 = self.hostz.copy()
devrec.copy_to_host(hostrec4)
np.testing.assert_equal(hostrec4, self.hostnz)
def test_device_record_copy(self):
hostrec = self.hostz.copy()
devrec = DeviceRecord(self.dtype)
devrec.copy_to_device(hostrec)
# Copy back and check values are all zeros
hostrec2 = self.hostnz.copy()
devrec.copy_to_host(hostrec2)
np.testing.assert_equal(self.hostz, hostrec2)
# Copy non-zero values to GPU and back and check values
hostrec3 = self.hostnz.copy()
devrec.copy_to_device(hostrec3)
hostrec4 = self.hostz.copy()
devrec.copy_to_host(hostrec4)
np.testing.assert_equal(hostrec4, self.hostnz)
def test_device_record_interface(self):
hostrec = self.hostz.copy()
devrec = DeviceRecord(self.dtype)
self._check_device_record(hostrec, devrec)