namespace mlir {
namespace transform {
class ApplyGPURewritePatternsOp;
}
}
namespace mlir {
namespace transform {
class ApplyGPUSubgroupReduceToNVVMConversionPatternsOp;
}
}
namespace mlir {
namespace transform {
class ApplyGPUToNVVMConversionPatternsOp;
}
}
namespace mlir {
namespace transform {
class ApplyGPUWwmaToNVVMConversionPatternsOp;
}
}
namespace mlir {
namespace transform {
class ApplyUnrollVectorsSubgroupMmaOp;
}
}
namespace mlir {
namespace transform {
class EliminateBarriersOp;
}
}
namespace mlir {
namespace transform {
class MapForallToBlocks;
}
}
namespace mlir {
namespace transform {
class MapNestedForallToThreads;
}
}
#ifdef GET_OP_CLASSES
#undef GET_OP_CLASSES
namespace mlir {
namespace transform {
namespace detail {
class ApplyGPURewritePatternsOpGenericAdaptorBase { … };
}
template <typename RangeT>
class ApplyGPURewritePatternsOpGenericAdaptor : public detail::ApplyGPURewritePatternsOpGenericAdaptorBase { … };
class ApplyGPURewritePatternsOpAdaptor : public ApplyGPURewritePatternsOpGenericAdaptor<::mlir::ValueRange> { … };
class ApplyGPURewritePatternsOp : public ::mlir::Op<ApplyGPURewritePatternsOp, ::mlir::OpTrait::ZeroRegions, ::mlir::OpTrait::ZeroResults, ::mlir::OpTrait::ZeroSuccessors, ::mlir::OpTrait::ZeroOperands, ::mlir::OpTrait::OpInvariants, ::mlir::transform::PatternDescriptorOpInterface::Trait> { … };
}
}
MLIR_DECLARE_EXPLICIT_TYPE_ID(::mlir::transform::ApplyGPURewritePatternsOp)
namespace mlir {
namespace transform {
namespace detail {
class ApplyGPUSubgroupReduceToNVVMConversionPatternsOpGenericAdaptorBase { … };
}
template <typename RangeT>
class ApplyGPUSubgroupReduceToNVVMConversionPatternsOpGenericAdaptor : public detail::ApplyGPUSubgroupReduceToNVVMConversionPatternsOpGenericAdaptorBase { … };
class ApplyGPUSubgroupReduceToNVVMConversionPatternsOpAdaptor : public ApplyGPUSubgroupReduceToNVVMConversionPatternsOpGenericAdaptor<::mlir::ValueRange> { … };
class ApplyGPUSubgroupReduceToNVVMConversionPatternsOp : public ::mlir::Op<ApplyGPUSubgroupReduceToNVVMConversionPatternsOp, ::mlir::OpTrait::ZeroRegions, ::mlir::OpTrait::ZeroResults, ::mlir::OpTrait::ZeroSuccessors, ::mlir::OpTrait::ZeroOperands, ::mlir::OpTrait::OpInvariants, ::mlir::transform::ConversionPatternDescriptorOpInterface::Trait> { … };
}
}
MLIR_DECLARE_EXPLICIT_TYPE_ID(::mlir::transform::ApplyGPUSubgroupReduceToNVVMConversionPatternsOp)
namespace mlir {
namespace transform {
namespace detail {
class ApplyGPUToNVVMConversionPatternsOpGenericAdaptorBase { … };
}
template <typename RangeT>
class ApplyGPUToNVVMConversionPatternsOpGenericAdaptor : public detail::ApplyGPUToNVVMConversionPatternsOpGenericAdaptorBase { … };
class ApplyGPUToNVVMConversionPatternsOpAdaptor : public ApplyGPUToNVVMConversionPatternsOpGenericAdaptor<::mlir::ValueRange> { … };
class ApplyGPUToNVVMConversionPatternsOp : public ::mlir::Op<ApplyGPUToNVVMConversionPatternsOp, ::mlir::OpTrait::ZeroRegions, ::mlir::OpTrait::ZeroResults, ::mlir::OpTrait::ZeroSuccessors, ::mlir::OpTrait::ZeroOperands, ::mlir::OpTrait::OpInvariants, ::mlir::transform::ConversionPatternDescriptorOpInterface::Trait> { … };
}
}
MLIR_DECLARE_EXPLICIT_TYPE_ID(::mlir::transform::ApplyGPUToNVVMConversionPatternsOp)
namespace mlir {
namespace transform {
namespace detail {
class ApplyGPUWwmaToNVVMConversionPatternsOpGenericAdaptorBase { … };
}
template <typename RangeT>
class ApplyGPUWwmaToNVVMConversionPatternsOpGenericAdaptor : public detail::ApplyGPUWwmaToNVVMConversionPatternsOpGenericAdaptorBase { … };
class ApplyGPUWwmaToNVVMConversionPatternsOpAdaptor : public ApplyGPUWwmaToNVVMConversionPatternsOpGenericAdaptor<::mlir::ValueRange> { … };
class ApplyGPUWwmaToNVVMConversionPatternsOp : public ::mlir::Op<ApplyGPUWwmaToNVVMConversionPatternsOp, ::mlir::OpTrait::ZeroRegions, ::mlir::OpTrait::ZeroResults, ::mlir::OpTrait::ZeroSuccessors, ::mlir::OpTrait::ZeroOperands, ::mlir::OpTrait::OpInvariants, ::mlir::transform::ConversionPatternDescriptorOpInterface::Trait> { … };
}
}
MLIR_DECLARE_EXPLICIT_TYPE_ID(::mlir::transform::ApplyGPUWwmaToNVVMConversionPatternsOp)
namespace mlir {
namespace transform {
namespace detail {
class ApplyUnrollVectorsSubgroupMmaOpGenericAdaptorBase { … };
}
template <typename RangeT>
class ApplyUnrollVectorsSubgroupMmaOpGenericAdaptor : public detail::ApplyUnrollVectorsSubgroupMmaOpGenericAdaptorBase { … };
class ApplyUnrollVectorsSubgroupMmaOpAdaptor : public ApplyUnrollVectorsSubgroupMmaOpGenericAdaptor<::mlir::ValueRange> { … };
class ApplyUnrollVectorsSubgroupMmaOp : public ::mlir::Op<ApplyUnrollVectorsSubgroupMmaOp, ::mlir::OpTrait::ZeroRegions, ::mlir::OpTrait::ZeroResults, ::mlir::OpTrait::ZeroSuccessors, ::mlir::OpTrait::ZeroOperands, ::mlir::OpTrait::OpInvariants, ::mlir::BytecodeOpInterface::Trait, ::mlir::transform::PatternDescriptorOpInterface::Trait> { … };
}
}
MLIR_DECLARE_EXPLICIT_TYPE_ID(::mlir::transform::ApplyUnrollVectorsSubgroupMmaOp)
namespace mlir {
namespace transform {
namespace detail {
class EliminateBarriersOpGenericAdaptorBase { … };
}
template <typename RangeT>
class EliminateBarriersOpGenericAdaptor : public detail::EliminateBarriersOpGenericAdaptorBase { … };
class EliminateBarriersOpAdaptor : public EliminateBarriersOpGenericAdaptor<::mlir::ValueRange> { … };
class EliminateBarriersOp : public ::mlir::Op<EliminateBarriersOp, ::mlir::OpTrait::ZeroRegions, ::mlir::OpTrait::ZeroResults, ::mlir::OpTrait::ZeroSuccessors, ::mlir::OpTrait::ZeroOperands, ::mlir::OpTrait::OpInvariants, ::mlir::transform::PatternDescriptorOpInterface::Trait> { … };
}
}
MLIR_DECLARE_EXPLICIT_TYPE_ID(::mlir::transform::EliminateBarriersOp)
namespace mlir {
namespace transform {
namespace detail {
class MapForallToBlocksGenericAdaptorBase { … };
}
template <typename RangeT>
class MapForallToBlocksGenericAdaptor : public detail::MapForallToBlocksGenericAdaptorBase { … };
class MapForallToBlocksAdaptor : public MapForallToBlocksGenericAdaptor<::mlir::ValueRange> { … };
class MapForallToBlocks : public ::mlir::Op<MapForallToBlocks, ::mlir::OpTrait::ZeroRegions, ::mlir::OpTrait::OneResult, ::mlir::OpTrait::OneTypedResult<::mlir::transform::TransformHandleTypeInterface>::Impl, ::mlir::OpTrait::ZeroSuccessors, ::mlir::OpTrait::OneOperand, ::mlir::OpTrait::OpInvariants, ::mlir::BytecodeOpInterface::Trait, ::mlir::transform::FunctionalStyleTransformOpTrait, ::mlir::MemoryEffectOpInterface::Trait, ::mlir::transform::TransformOpInterface::Trait, ::mlir::transform::TransformEachOpTrait> { … };
}
}
MLIR_DECLARE_EXPLICIT_TYPE_ID(::mlir::transform::MapForallToBlocks)
namespace mlir {
namespace transform {
namespace detail {
class MapNestedForallToThreadsGenericAdaptorBase { … };
}
template <typename RangeT>
class MapNestedForallToThreadsGenericAdaptor : public detail::MapNestedForallToThreadsGenericAdaptorBase { … };
class MapNestedForallToThreadsAdaptor : public MapNestedForallToThreadsGenericAdaptor<::mlir::ValueRange> { … };
class MapNestedForallToThreads : public ::mlir::Op<MapNestedForallToThreads, ::mlir::OpTrait::ZeroRegions, ::mlir::OpTrait::OneResult, ::mlir::OpTrait::OneTypedResult<::mlir::transform::TransformHandleTypeInterface>::Impl, ::mlir::OpTrait::ZeroSuccessors, ::mlir::OpTrait::OneOperand, ::mlir::OpTrait::OpInvariants, ::mlir::BytecodeOpInterface::Trait, ::mlir::transform::FunctionalStyleTransformOpTrait, ::mlir::MemoryEffectOpInterface::Trait, ::mlir::transform::TransformEachOpTrait, ::mlir::transform::TransformOpInterface::Trait> { … };
}
}
MLIR_DECLARE_EXPLICIT_TYPE_ID(::mlir::transform::MapNestedForallToThreads)
#endif