Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Upstream changes in bin/triton-tensor-layout.cpp #2904

Closed
whitneywhtsang opened this issue Dec 3, 2024 · 2 comments
Closed

Upstream changes in bin/triton-tensor-layout.cpp #2904

whitneywhtsang opened this issue Dec 3, 2024 · 2 comments
Assignees
Labels
enhancement New feature or request upstream: triton

Comments

@whitneywhtsang
Copy link
Contributor

whitneywhtsang commented Dec 3, 2024

The following changes were made in 816d7ef.

diff --git a/bin/triton-tensor-layout.cpp b/bin/triton-tensor-layout.cpp
index 7c635dafa..49a07681e 100644
--- a/bin/triton-tensor-layout.cpp
+++ b/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 == "ttg") {
-    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();
 }
@chengjunlu
Copy link
Contributor

The upstream PR triton-lang/triton#5444

@whitneywhtsang
Copy link
Contributor Author

Merged in #3023.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request upstream: triton
Projects
None yet
Development

No branches or pull requests

3 participants