More progress on SCF type conversion

2022-05-05 20:56:55 +08:00
parent 26c59e4718
commit 0c5319eed9
2 changed files with 40 additions and 11 deletions
--- a/lib/Dialect/TritonGPU/Transforms/TritonGPUConversion.cpp
+++ b/lib/Dialect/TritonGPU/Transforms/TritonGPUConversion.cpp
@@ -50,7 +50,9 @@ TritonGPUTypeConverter::TritonGPUTypeConverter(MLIRContext *context,
  // materailizations
  addArgumentMaterialization([&](OpBuilder &builder, RankedTensorType tensorType,
                                 ValueRange inputs, Location loc) {
-    // llvm::errs() << "Trying to materialize target... " << inputs[0] << "\n";
+    llvm::errs() << "Trying to materialize target... " << inputs[0] << "\n"
+                 << "in: \n";
+    inputs[0].dyn_cast<BlockArgument>().getOwner()->getParentOp()->getParentOp()->print(llvm::errs());
    llvm_unreachable("Not implemented");
    return llvm::None;
  });