@@ -2995,20 +2995,20 @@ def NVVM_WgmmaMmaAsyncOp : NVVM_Op<"wgmma.mma_async",
2995
2995
// NVVM Griddepcontrol Ops
2996
2996
//===----------------------------------------------------------------------===//
2997
2997
2998
- def FDLWait : I32EnumAttrCase<"wait", 0>;
2999
- def FDLLaunchDependent : I32EnumAttrCase<"launch_dependents", 1>;
2998
+ def GridDepActionWait : I32EnumAttrCase<"wait", 0>;
2999
+ def GridDepActionLaunchDependent : I32EnumAttrCase<"launch_dependents", 1>;
3000
3000
3001
- def FDLKind : I32EnumAttr<"FDLKind ", "Fast dependenct launch kind ",
3002
- [FDLWait, FDLLaunchDependent ]> {
3001
+ def GridDepActionKind : I32EnumAttr<"GridDepKind ", "Action kind for grid dependency control ",
3002
+ [GridDepActionWait, GridDepActionLaunchDependent ]> {
3003
3003
let genSpecializedAttr = 0;
3004
3004
let cppNamespace = "::mlir::NVVM";
3005
3005
}
3006
3006
3007
- def FDLKindAttr : EnumAttr<NVVM_Dialect, FDLKind , "fdl_kind ">;
3007
+ def GridDepActionAttr : EnumAttr<NVVM_Dialect, GridDepActionKind , "grid_dep_action ">;
3008
3008
3009
3009
def NVVM_GriddepcontrolOp : NVVM_Op<"griddepcontrol", []> {
3010
3010
let description = [{
3011
- If the $kind attribute is set to `wait`, the instruction, it causes the
3011
+ If the $kind attribute is set to `wait`, it causes the
3012
3012
executing thread to wait until all prerequisite grids in flight
3013
3013
have completed and all the memory operations from the prerequisite grids
3014
3014
are performed and made visible to the current grid.
@@ -3021,11 +3021,11 @@ def NVVM_GriddepcontrolOp : NVVM_Op<"griddepcontrol", []> {
3021
3021
[For more information, see PTX ISA](https://docs.nvidia.com/cuda/parallel-thread-execution/#parallel-synchronization-and-communication-instructions-griddepcontrol)
3022
3022
}];
3023
3023
3024
- let arguments = (ins FDLKindAttr :$kind);
3024
+ let arguments = (ins GridDepActionAttr :$kind);
3025
3025
3026
3026
let assemblyFormat = "$kind attr-dict";
3027
3027
3028
- string llvmBuilder = [{
3028
+ string llvmBuilder = [{
3029
3029
llvm::Intrinsic::ID id;
3030
3030
switch ($kind) {
3031
3031
case NVVM::FDLKind::wait:
0 commit comments