def testSample(self, check_nvidia_smi_exists: mock.Mock) -> None:
samples = cuda_memcopy_benchmark.Run(mock.Mock(vms=[self.MockVm()]))
expected = sample.Sample(
'H2D-Pinned', 8494.3, 'MB/s', {
'time':
0.00377,
'size':
33554432,
'NumDevsUsed':
'1',
'device':
0,
'command':
'/usr/local/cuda/extras/demo_suite/bandwidthTest --csv '
'--memory=pinned --mode=quick --htod --dtoh --dtod --device=0',
'memory':
'pinned',
'mode':
'quick',
'htod':
True,
'dtoh':
True,
'dtod':
True,
'wc':
False,
})
self.assertSamplesEqualUpToTimestamp(expected, samples[0])
def testCmd(self, check_nvidia_smi_exists: mock.Mock) -> None:
vm = self.MockVm()
cuda_memcopy_benchmark.Run(mock.Mock(vms=[vm]))
vm.RemoteCommandWithReturnCode.assert_called_with(
'/usr/local/cuda/extras/demo_suite/bandwidthTest --csv --memory=pinned '
'--mode=quick --htod --dtoh --dtod --device=0',
ignore_failure=True)
def testEmptySample(self, check_nvidia_smi_exists: mock.Mock) -> None:
samples = cuda_memcopy_benchmark.Run(mock.Mock(vms=[self.MockVm()]))
self.assertLen(samples, 0)