[PYTHON][OPS] Fixed typo in einsum

2020-02-25 16:37:55 -05:00
parent 646c49f847
commit ea37ba5d35
1 changed files with 1 additions and 1 deletions
--- 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;