gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[FRONTEND] add an attr for masked load without explicit other (#55)

Browse Source
This commit is contained in:
Shintaro Iwasaki
2022-08-18 09:51:37 -07:00
committed by GitHub
parent fc58250a06
commit d69ce77b19
16 changed files with 71 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
include/triton/Dialect/Triton/IR/TritonOps.td
View File

@@ -80,7 +80,7 @@ def TT_LoadOp : TT_Op<"load",
let arguments = (ins TT_PtrTensor:$ptr, BoolLike:$mask, TT_Type:$other,
TT_CacheModifierAttr:$cache, TT_EvictionPolicyAttr:$evict,
BoolAttr:$isVolatile);
BoolAttr:$isVolatile, BoolAttr:$isOtherUnspecified);
let results = (outs TT_Type:$result);
@@ -89,7 +89,8 @@ def TT_LoadOp : TT_Op<"load",
OpBuilder<(ins "Value":$ptr, "triton::CacheModifier":$cache,
"triton::EvictionPolicy":$evict, "bool":$isVolatile)>,
OpBuilder<(ins "Value":$ptr, "Value":$mask, "triton::CacheModifier":$cache,
"triton::EvictionPolicy":$evict, "bool":$isVolatile)>
"triton::EvictionPolicy":$evict, "bool":$isVolatile,
"bool":$isOtherUnspecified)>
];
let assemblyFormat = "$ptr`,` $mask`,` $other attr-dict `:` type($result)";

3
include/triton/Dialect/TritonGPU/IR/TritonGPUOps.td
View File

@@ -48,7 +48,8 @@ def TTG_CopyAsyncOp : TTG_Op<"copy_async",
let arguments = (ins TT_PtrTensor:$ptr, BoolLike:$mask, TT_Type:$other,
TT_CacheModifierAttr:$cache, TT_EvictionPolicyAttr:$evict,
BoolAttr:$isVolatile);
BoolAttr:$isVolatile,
BoolAttr:$isOtherUnspecified);
let results = (outs TT_Type:$result);