Now vecadd works

2022-03-30 20:21:47 +08:00
parent e381dc72c5
commit 2041b67fbf
5 changed files with 285 additions and 386 deletions
--- a/include/triton/ir/TritonOps.td
+++ b/include/triton/ir/TritonOps.td
@@ -114,7 +114,9 @@ def TT_EvictionPolicyAttr : I32EnumAttr<
 def TT_LoadOp : TT_Op<"load", [SameOperandsAndResultShape]> {
    let summary = "load";

-    let arguments = (ins TT_PtrTensor:$ptr, BoolLike:$mask, TT_Type:$other);
+    let arguments = (ins TT_PtrTensor:$ptr, BoolLike:$mask, TT_Type:$other,
+                         TT_CacheModifierAttr:$cache, TT_EvictionPolicyAttr:$evict,
+                         BoolAttr:$isVolatile);

    let results = (outs TT_Type:$result);