Add tt.trans op support (#126)

Author: chengjunlu

lib/Conversion/TritonGPUToSPIRV/ViewOpToSPIRV.cpp

@@ -205,30 +205,28 @@ struct ExpandDimsOpSPIRVConversion
   }
 };
 
-#if 0
-struct TransOpConversion
-        : public ConvertTritonGPUOpToLLVMPattern<triton::TransOp> {
-  using ConvertTritonGPUOpToLLVMPattern<
-          triton::TransOp>::ConvertTritonGPUOpToLLVMPattern;
+struct TransOpSPIRVConversion
+    : public ConvertTritonGPUOpToSPIRVPattern<triton::TransOp> {
+  using ConvertTritonGPUOpToSPIRVPattern<
+      triton::TransOp>::ConvertTritonGPUOpToSPIRVPattern;
 
   LogicalResult
   matchAndRewrite(triton::TransOp op, OpAdaptor adaptor,
                   ConversionPatternRewriter &rewriter) const override {
     Location loc = op->getLoc();
     auto srcSmemObj =
-            getSharedMemoryObjectFromStruct(loc, adaptor.getSrc(), rewriter);
+        getSharedMemoryObjectFromStruct(loc, adaptor.getSrc(), rewriter);
     SmallVector<Value> dstStrides = {srcSmemObj.strides[1],
                                      srcSmemObj.strides[0]};
     SmallVector<Value> dstOffsets = {srcSmemObj.offsets[1],
                                      srcSmemObj.offsets[0]};
     auto dstSmemObj =
-            SharedMemoryObject(srcSmemObj.base, dstStrides, dstOffsets);
+        SharedMemoryObject(srcSmemObj.base, dstStrides, dstOffsets);
     auto retVal = getStructFromSharedMemoryObject(loc, dstSmemObj, rewriter);
     rewriter.replaceOp(op, retVal);
     return success();
   }
 };
-#endif
 
 void populateViewOpToSPIRVPatterns(
     TritonGPUToSPIRVTypeConverter &typeConverter, mlir::MLIRContext *context,
@@ -245,4 +243,5 @@ void populateViewOpToSPIRVPatterns(
       mlir::spirv::checkOpSupported(computeCapability,
                                     "INTELConvertFToBF16Op"));
   patterns.add<CatOpSPIRVConversion>(typeConverter, context, benefit);
+  patterns.add<TransOpSPIRVConversion>(typeConverter, context, benefit);
 }