def test_to_device_to_host(self): """ Tests .to_device() and .copy_to_host() """ n = 10 data = np.zeros(n) output = np.zeros(n) @vectorize("float64(float64)", target='roc') def func(x): return x + 1 hsaapi.to_device(data) out_device = hsaapi.to_device(output) func(data, out=out_device) host_output = out_device.copy_to_host() np.testing.assert_equal(np.ones(n), host_output)
def to_device(self, hostary): if dgpu_present: return api.to_device(hostary) else: return hostary
def to_device(self, hostary, stream): if dgpu_present: return api.to_device(hostary) else: return hostary