[CODEGEN][ANALYSIS] Fixed issue in layout inference

2020-08-07 00:49:04 -04:00
parent 89e456107b
commit ba9955ae39
9 changed files with 109 additions and 72 deletions
--- a/python/examples/tutorials/mat_mul.py
+++ b/python/examples/tutorials/mat_mul.py
@@ -112,7 +112,7 @@ class _dot(torch.autograd.Function):
        time = kernel(a, b, c, 1., M, N, K, 
                      a.stride(0), b.stride(0), c.stride(0), 
                      grid=grid, bench=100)
-        print(time)
+        print(2*M*N*K/(time*1e-6)*1e-9)
        return c