From 801c8a4c924f58bf15e9ad8d7270f00e4158478a Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Sun, 5 Jun 2022 12:32:07 -0700 Subject: [PATCH] [TUTORIALS] Fixed typo --- python/tutorials/03-matrix-multiplication.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tutorials/03-matrix-multiplication.py b/python/tutorials/03-matrix-multiplication.py index 912833c52..2dfa98a42 100644 --- a/python/tutorials/03-matrix-multiplication.py +++ b/python/tutorials/03-matrix-multiplication.py @@ -297,7 +297,7 @@ def matmul(a, b, activation=None): torch.manual_seed(0) a = torch.randn((512, 512), device='cuda', dtype=torch.float16) b = torch.randn((512, 512), device='cuda', dtype=torch.float16) -triton_output = matmul(a, b, activation=leaky_relu) +triton_output = matmul(a, b) torch_output = torch.matmul(a, b) print(f"triton_output={triton_output}") print(f"torch_output={torch_output}")