small fixes

Author: Jianhui-Li

mlir/include/mlir/Dialect/XeGPU/IR/XeGPUOps.td

@@ -665,7 +665,6 @@ def XeGPU_PrefetchOp : XeGPU_Op<"prefetch", []> {
     prop-dict
     attr-dict `:` type(operands) 
   }];
-    // type($source) (type($offsets)^)?
 
   let builders = [
     OpBuilder<(ins "Value": $source,
@@ -763,10 +762,6 @@ def XeGPU_LoadGatherOp : XeGPU_Op<"load", [MemoryEffects<[MemRead]>]> {
     $mask prop-dict 
     attr-dict `:` type(operands) `->` type($value)
   }];
-  
-  //    functional-type(operands, results)
-  // type($source) (type($offsets)^ )? `,` type($mask) `->` type($value)
-  
 
   let builders = [
     OpBuilder<(ins "Type": $value, "Value": $source, "Value": $mask,
@@ -855,7 +850,6 @@ def XeGPU_StoreScatterOp : XeGPU_Op<"store", [MemoryEffects<[MemWrite]>]> {
     prop-dict 
     attr-dict `:`  type(operands)
   }];
-//    type($value) `,` qualified(type($dest)) (type($offsets)^)? `,` type($mask)
 
   let builders = [
     OpBuilder<(ins "Value": $value, "Value": $dest, "Value": $mask,

mlir/lib/Dialect/XeGPU/Transforms/XeGPUUnroll.cpp

@@ -502,7 +502,7 @@ struct UnrollLoadGatherOp : public UnrollPattern<xegpu::LoadGatherOp> {
     SmallVector<Type> convertedTdescTypes =
         getUnrolledTypes(tdescTy, *targetShape);
     SmallVector<Value> convertedTdescs = pack(
-        op.getSource(), convertedTdescTypes, *targetShape, loc, rewriter);
+        op.getTensorDesc(), convertedTdescTypes, *targetShape, loc, rewriter);
 
     SmallVector<Type> convertedMaskTypes;
     SmallVector<Value> convertedMasks;

mlir/test/Dialect/XeGPU/invalid.mlir

@@ -405,7 +405,7 @@ func.func @load_gather_offset_wi(%src: ui64) {
 }
 
 // -----
-func.func @store_scatter_offset(%src: memref<?xf16>) {
+func.func @store_scatter_offset_sg(%src: memref<?xf16>) {
   %val = arith.constant dense<2.9>: vector<4xf16>
   %offsets = arith.constant dense<[0]> : vector<1xindex>
   %mask = arith.constant dense<1>: vector<1xi1>