OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 00:20:20 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Change to only run one round of matmul benchmark.

Browse Source 
PiperOrigin-RevId: 163364341
This commit is contained in:
Yangzihao Wang 2017-07-27 11:08:47 -07:00 committed by TensorFlower Gardener
parent ae3119d16b
commit 3b97f1f9bd
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/ops/matmul_benchmark.py
View File

@ -106,7 +106,7 @@ class MatmulBenchmark(test.Benchmark):
device=device, device=device,
dtype=str(dtype).replace(' ', ''), dtype=str(dtype).replace(' ', ''),
inputinfo=str(n) + 'x' + str(m) + 'x' + str(k) + ',ta:' + inputinfo=str(n) + 'x' + str(m) + 'x' + str(k) + ',ta:' +
str(transpose_a) + '.tb:' + str(transpose_b)).replace(' ', ''), str(transpose_a) + ',tb:' + str(transpose_b)).replace(' ', ''),
iters=num_iters, iters=num_iters,
wall_time=duration) wall_time=duration)
return duration return duration
@ -134,9 +134,7 @@ class MatmulBenchmark(test.Benchmark):
self.run_test_gpu(n, m, k, transpose_a, transpose_b, dtype, num_iters) self.run_test_gpu(n, m, k, transpose_a, transpose_b, dtype, num_iters)
def benchmark_matmul(self): def benchmark_matmul(self):
num_iters = 200 self.test_round(num_iters=200)
for _ in range(10):
self.test_round(num_iters)
if __name__ == '__main__': if __name__ == '__main__':