From 52af8cda34dd4166395f27ffa37af23f05d3e939 Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Sun, 31 Jan 2021 13:36:19 -0500 Subject: [PATCH] [PYTHON] Fixed issue with IS_TK_DIV_K --- python/triton/ops/matmul.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/triton/ops/matmul.py b/python/triton/ops/matmul.py index f3a1348e0..c7fcfd2a2 100644 --- a/python/triton/ops/matmul.py +++ b/python/triton/ops/matmul.py @@ -57,7 +57,7 @@ class _matmul(torch.autograd.Function): 'TN' : _matmul.TN, 'TK' : _matmul.TK, 'TZ' : _matmul.TZ, - 'IS_TK_DIV_K' : is_tk_div_k + 'IS_TK_DIV_K' : int(is_tk_div_k) } _matmul._kernels[key] = triton.kernel(_matmul.src, device, num_warps=_matmul.num_warps, defines=defines) kernel = _matmul._kernels[key]