================ @@ -20,6 +20,7 @@ #ifndef NVGPU #define NVGPU +include "mlir/Interfaces/InferTypeOpInterface.td" ---------------- Observer007 wrote: Great, `PredOpTrait` and `SameOperandsAndResultShape` successfully solved my question. Thanks a lot. https://github.com/llvm/llvm-project/pull/100965