@@ -976,7 +976,7 @@ def OpenACC_ParallelOp : OpenACC_Op<"parallel",
976
976
`)`
977
977
| `vector_length` `(` custom<DeviceTypeOperands>($vectorLength,
978
978
type($vectorLength), $vectorLengthDeviceType) `)`
979
- | `wait` `(` custom<WaitOperands >($waitOperands,
979
+ | `wait` `(` custom<DeviceTypeOperandsWithSegment >($waitOperands,
980
980
type($waitOperands), $waitOperandsDeviceType, $waitOperandsSegments) `)`
981
981
| `self` `(` $selfCond `)`
982
982
| `if` `(` $ifCond `)`
@@ -1075,7 +1075,7 @@ def OpenACC_SerialOp : OpenACC_Op<"serial",
1075
1075
| `private` `(` custom<SymOperandList>(
1076
1076
$gangPrivateOperands, type($gangPrivateOperands), $privatizations)
1077
1077
`)`
1078
- | `wait` `(` custom<WaitOperands >($waitOperands,
1078
+ | `wait` `(` custom<DeviceTypeOperandsWithSegment >($waitOperands,
1079
1079
type($waitOperands), $waitOperandsDeviceType, $waitOperandsSegments) `)`
1080
1080
| `self` `(` $selfCond `)`
1081
1081
| `if` `(` $ifCond `)`
@@ -1196,8 +1196,9 @@ def OpenACC_KernelsOp : OpenACC_Op<"kernels",
1196
1196
type($numWorkers), $numWorkersDeviceType) `)`
1197
1197
| `vector_length` `(` custom<DeviceTypeOperands>($vectorLength,
1198
1198
type($vectorLength), $vectorLengthDeviceType) `)`
1199
- | `wait` `(` custom<WaitOperands>($waitOperands,
1200
- type($waitOperands), $waitOperandsDeviceType, $waitOperandsSegments) `)`
1199
+ | `wait` `(` custom<DeviceTypeOperandsWithSegment>($waitOperands,
1200
+ type($waitOperands), $waitOperandsDeviceType,
1201
+ $waitOperandsSegments) `)`
1201
1202
| `self` `(` $selfCond `)`
1202
1203
| `if` `(` $ifCond `)`
1203
1204
)
0 commit comments