test: print transfer rates in MiB/s

This commit is contained in:
Sebastien Bourdeauducq 2020-07-07 17:28:15 +08:00
parent 13501115f6
commit 85b5a04acf
1 changed files with 3 additions and 3 deletions

View File

@ -52,19 +52,19 @@ class TransferTest(ExperimentCase):
def test_host_to_device(self): def test_host_to_device(self):
exp = self.create(_Transfer) exp = self.create(_Transfer)
host_to_device_rate = exp.host_to_device() host_to_device_rate = exp.host_to_device()
print(host_to_device_rate, "B/s") print(host_to_device_rate/(1024*1024), "MiB/s")
self.assertGreater(host_to_device_rate, 2.0e6) self.assertGreater(host_to_device_rate, 2.0e6)
def test_device_to_host(self): def test_device_to_host(self):
exp = self.create(_Transfer) exp = self.create(_Transfer)
device_to_host_rate = exp.device_to_host() device_to_host_rate = exp.device_to_host()
print(device_to_host_rate, "B/s") print(device_to_host_rate/(1024*1024), "MiB/s")
self.assertGreater(device_to_host_rate, 2.3e6) self.assertGreater(device_to_host_rate, 2.3e6)
def test_device_to_host_array(self): def test_device_to_host_array(self):
exp = self.create(_Transfer) exp = self.create(_Transfer)
rate = exp.device_to_host_array() rate = exp.device_to_host_array()
print(rate, "B/s") print(rate/(1024*1024), "MiB/s")
self.assertGreater(rate, .15e6) self.assertGreater(rate, .15e6)