From 51bc244b2c4f761fe0e4af82e98070b568fd7fac Mon Sep 17 00:00:00 2001
From: Philippe Tillet <ptillet@g.harvard.edu>
Date: Tue, 25 Feb 2020 16:37:55 -0500
Subject: [PATCH] [PYTHON][OPS] Fixed typo in einsum

---
 python/triton/ops/einsum.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/triton/ops/einsum.py b/python/triton/ops/einsum.py
index a24e422dc..bda5d3b16 100644
--- a/python/triton/ops/einsum.py
+++ b/python/triton/ops/einsum.py
@@ -247,7 +247,7 @@ __global__ void {name}(
         acc += a @ b;
         #ifdef MASK
         uint32 bits[TM, TN, TB] = bitcast<uint32[TM,TN,TB]>(acc);
-        acc = bitcast<TYPE[TM, TN, TB]>(bits & MASK);
+        acc = bitcast<float[TM, TN, TB]>(bits & MASK);
         #endif
 
         checkk = k > TK;