Enable the triton-tensor-layout utils for TritonIntelGPU dialect.

Author: chengjunlu

bin/CMakeLists.txt

@@ -102,6 +102,7 @@ add_llvm_executable(triton-tensor-layout triton-tensor-layout.cpp PARTIAL_SOURCE
 target_link_libraries(triton-tensor-layout PRIVATE
   TritonGPUIR
   TritonNvidiaGPUIR
+  TritonIntelGPUIR
   ${triton_libs}
   ${conversion_libs}
   ${dialect_libs}

bin/triton-tensor-layout.cpp

@@ -80,17 +80,9 @@ static cl::opt<std::string> TensorStr(
 //===--------------------------------------------------------------------===//
 
 LogicalResult layoutPrint(RankedTensorType tensorType, raw_ostream &os) {
-  StringRef dialectName = tensorType.getEncoding().getDialect().getNamespace();
-
   // Dispatch to the corresponding dialect helper function to print the layout.
-  if (dialectName == "triton_gpu") {
-    os << triton::gpu::getLayoutStr(tensorType, UseHWPointOfView);
-    return success();
-  }
-
-  llvm::errs() << "Unsupported tensor layout attribute: "
-               << tensorType.getEncoding() << "\n";
-  return failure();
+  os << triton::gpu::getLayoutStr(tensorType, UseHWPointOfView);
+  return success();
 }
 
 LogicalResult printLayoutFromFile(MLIRContext *context, StringRef filename,