@@ -976,7 +976,7 @@ def OpenACC_ParallelOp : OpenACC_Op<"parallel",
976976 `)`
977977 | `vector_length` `(` custom<DeviceTypeOperands>($vectorLength,
978978 type($vectorLength), $vectorLengthDeviceType) `)`
979- | `wait` `(` custom<WaitOperands >($waitOperands,
979+ | `wait` `(` custom<DeviceTypeOperandsWithSegment >($waitOperands,
980980 type($waitOperands), $waitOperandsDeviceType, $waitOperandsSegments) `)`
981981 | `self` `(` $selfCond `)`
982982 | `if` `(` $ifCond `)`
@@ -1075,7 +1075,7 @@ def OpenACC_SerialOp : OpenACC_Op<"serial",
10751075 | `private` `(` custom<SymOperandList>(
10761076 $gangPrivateOperands, type($gangPrivateOperands), $privatizations)
10771077 `)`
1078- | `wait` `(` custom<WaitOperands >($waitOperands,
1078+ | `wait` `(` custom<DeviceTypeOperandsWithSegment >($waitOperands,
10791079 type($waitOperands), $waitOperandsDeviceType, $waitOperandsSegments) `)`
10801080 | `self` `(` $selfCond `)`
10811081 | `if` `(` $ifCond `)`
@@ -1196,8 +1196,9 @@ def OpenACC_KernelsOp : OpenACC_Op<"kernels",
11961196 type($numWorkers), $numWorkersDeviceType) `)`
11971197 | `vector_length` `(` custom<DeviceTypeOperands>($vectorLength,
11981198 type($vectorLength), $vectorLengthDeviceType) `)`
1199- | `wait` `(` custom<WaitOperands>($waitOperands,
1200- type($waitOperands), $waitOperandsDeviceType, $waitOperandsSegments) `)`
1199+ | `wait` `(` custom<DeviceTypeOperandsWithSegment>($waitOperands,
1200+ type($waitOperands), $waitOperandsDeviceType,
1201+ $waitOperandsSegments) `)`
12011202 | `self` `(` $selfCond `)`
12021203 | `if` `(` $ifCond `)`
12031204 )
0 commit comments