================ @@ -20,6 +20,7 @@ #ifndef NVGPU #define NVGPU +include "mlir/Interfaces/InferTypeOpInterface.td" ---------------- grypp wrote: What about `PredOpTrait`? https://github.com/llvm/llvm-project/pull/100965