xegpu: drop XeGPU prefix from transform op names

Author: tkarna

mlir/include/mlir/Dialect/XeGPU/TransformOps/XeGPUTransformOps.td

@@ -15,7 +15,7 @@ include "mlir/Dialect/Transform/IR/TransformTypes.td"
 include "mlir/IR/OpBase.td"
 include "mlir/Interfaces/SideEffectInterfaces.td"
 
-def XeGPUHoistDescOp : Op<Transform_Dialect, "xegpu.hoist_desc_ops", [
+def HoistDescOp : Op<Transform_Dialect, "xegpu.hoist_desc_ops", [
   TransformOpInterface, TransformEachOpTrait,
   DeclareOpInterfaceMethods<MemoryEffectsOpInterface>
 ]> {
@@ -41,7 +41,7 @@ def XeGPUHoistDescOp : Op<Transform_Dialect, "xegpu.hoist_desc_ops", [
   }];
 }
 
-def XeGPUSetDPASLayoutOp : Op<Transform_Dialect, "xegpu.set_dpas_layout", [
+def SetDPASLayoutOp : Op<Transform_Dialect, "xegpu.set_dpas_layout", [
   TransformOpInterface, TransformEachOpTrait,
   DeclareOpInterfaceMethods<MemoryEffectsOpInterface>
 ]> {
@@ -78,7 +78,7 @@ def XeGPUSetDPASLayoutOp : Op<Transform_Dialect, "xegpu.set_dpas_layout", [
   }];
 }
 
-def XeGPUInsertPrefetchOp : Op<Transform_Dialect, "xegpu.insert_prefetch",
+def InsertPrefetchOp : Op<Transform_Dialect, "xegpu.insert_prefetch",
   [FunctionalStyleTransformOpTrait, MemoryEffectsOpInterface,
    DeclareOpInterfaceMethods<TransformOpInterface>]> {
 
@@ -104,7 +104,7 @@ def XeGPUInsertPrefetchOp : Op<Transform_Dialect, "xegpu.insert_prefetch",
 // TODO this should be handled with gpu transform ops.
 // Add gpu mapping to scf.forall op and use something like
 // transform.gpu.map_forall_to_blocks to convert to gpu.launch op.
-def XeGPUSetGPULaunchThreadsOp
+def SetGPULaunchThreadsOp
     : Op<Transform_Dialect, "xegpu.set_gpu_launch_threads", [
       TransformOpInterface, TransformEachOpTrait,
       DeclareOpInterfaceMethods<MemoryEffectsOpInterface>

mlir/lib/Dialect/XeGPU/TransformOps/XeGPUTransformOps.cpp

@@ -413,7 +413,7 @@ scf::ForOp fuseForLoops(scf::ForOp target, scf::ForOp source,
   return fusedLoop;
 }
 
-DiagnosedSilenceableFailure transform::XeGPUHoistDescOp::applyToOne(
+DiagnosedSilenceableFailure transform::HoistDescOp::applyToOne(
     transform::TransformRewriter &rewriter, Operation *target,
     transform::ApplyToEachResultList &results,
     transform::TransformState &state) {
@@ -440,15 +440,15 @@ DiagnosedSilenceableFailure transform::XeGPUHoistDescOp::applyToOne(
   return DiagnosedSilenceableFailure::success();
 }
 
-void transform::XeGPUHoistDescOp::getEffects(
+void transform::HoistDescOp::getEffects(
     ::llvm::SmallVectorImpl<MemoryEffects::EffectInstance> &effects) {
   consumesHandle(getLoopMutable(), effects);
   producesHandle(getOperation()->getOpResults(), effects);
   modifiesPayload(effects);
 }
 
 DiagnosedSilenceableFailure
-transform::XeGPUInsertPrefetchOp::apply(transform::TransformRewriter &rewriter,
+transform::InsertPrefetchOp::apply(transform::TransformRewriter &rewriter,
                                         transform::TransformResults &results,
                                         transform::TransformState &state) {
 
@@ -573,7 +573,7 @@ transform::XeGPUInsertPrefetchOp::apply(transform::TransformRewriter &rewriter,
   return DiagnosedSilenceableFailure::success();
 }
 
-DiagnosedSilenceableFailure transform::XeGPUSetDPASLayoutOp::applyToOne(
+DiagnosedSilenceableFailure transform::SetDPASLayoutOp::applyToOne(
     transform::TransformRewriter &rewriter, Operation *target,
     transform::ApplyToEachResultList &results,
     transform::TransformState &state) {
@@ -668,13 +668,13 @@ DiagnosedSilenceableFailure transform::XeGPUSetDPASLayoutOp::applyToOne(
   return DiagnosedSilenceableFailure::success();
 }
 
-void transform::XeGPUSetDPASLayoutOp::getEffects(
+void transform::SetDPASLayoutOp::getEffects(
     ::llvm::SmallVectorImpl<MemoryEffects::EffectInstance> &effects) {
   onlyReadsHandle(getDpasOpMutable(), effects);
   modifiesPayload(effects);
 }
 
-DiagnosedSilenceableFailure transform::XeGPUSetGPULaunchThreadsOp::applyToOne(
+DiagnosedSilenceableFailure transform::SetGPULaunchThreadsOp::applyToOne(
     transform::TransformRewriter &rewriter, Operation *target,
     transform::ApplyToEachResultList &results,
     transform::TransformState &state) {
@@ -705,7 +705,7 @@ DiagnosedSilenceableFailure transform::XeGPUSetGPULaunchThreadsOp::applyToOne(
   return DiagnosedSilenceableFailure::success();
 }
 
-void transform::XeGPUSetGPULaunchThreadsOp::getEffects(
+void transform::SetGPULaunchThreadsOp::getEffects(
     ::llvm::SmallVectorImpl<MemoryEffects::EffectInstance> &effects) {
   onlyReadsHandle(getLaunchOpMutable(), effects);
   modifiesPayload(effects);

mlir/python/mlir/dialects/transform/xegpu.py

@@ -17,8 +17,8 @@
 
 
 @_ods_cext.register_operation(_Dialect, replace=True)
-class XeGPUSetDPASLayoutOp(XeGPUSetDPASLayoutOp):
-    """Specialization for XeGPUSetDPASLayoutOp class."""
+class SetDPASLayoutOp(SetDPASLayoutOp):
+    """Specialization for SetDPASLayoutOp class."""
 
     def __init__(
         self,
@@ -45,8 +45,8 @@ def __init__(
 
 
 @_ods_cext.register_operation(_Dialect, replace=True)
-class XeGPUInsertPrefetchOp(XeGPUInsertPrefetchOp):
-    """Specialization for XeGPUInsertPrefetchOp class."""
+class InsertPrefetchOp(InsertPrefetchOp):
+    """Specialization for InsertPrefetchOp class."""
 
     def __init__(
         self,
@@ -75,8 +75,8 @@ def __init__(
 
 
 @_ods_cext.register_operation(_Dialect, replace=True)
-class XeGPUHoistDescOp(XeGPUHoistDescOp):
-    """Specialization for XeGPUHoistDescOp class."""
+class HoistDescOp(HoistDescOp):
+    """Specialization for HoistDescOp class."""
 
     def __init__(
         self,
@@ -94,8 +94,8 @@ def __init__(
 
 
 @_ods_cext.register_operation(_Dialect, replace=True)
-class XeGPUSetGPULaunchThreadsOp(XeGPUSetGPULaunchThreadsOp):
-    """Specialization for XeGPUSetGPULaunchThreadsOp class."""
+class SetGPULaunchThreadsOp(SetGPULaunchThreadsOp):
+    """Specialization for SetGPULaunchThreadsOp class."""
 
     def __init__(
         self,