diff --git a/test/TLX/print-ttgir-to-tlx.mlir b/test/TLX/print-ttgir-to-tlx.mlir index 715a5bcd64..f4cabff336 100644 --- a/test/TLX/print-ttgir-to-tlx.mlir +++ b/test/TLX/print-ttgir-to-tlx.mlir @@ -36,6 +36,7 @@ // CHECK-DAG: tlx.tmem_store( // CHECK-DAG: tlx.local_trans( // CHECK-DAG: tlx.subslice( +// CHECK-DAG: tlx.local_view( // Verify warp specialization uses Python-like async_tasks syntax // CHECK-DAG: with tlx.async_tasks(): diff --git a/third_party/tlx/dialect/lib/Transforms/PrintTTGIRToTLX.cpp b/third_party/tlx/dialect/lib/Transforms/PrintTTGIRToTLX.cpp index c22fd8396a..15dfc0f166 100644 --- a/third_party/tlx/dialect/lib/Transforms/PrintTTGIRToTLX.cpp +++ b/third_party/tlx/dialect/lib/Transforms/PrintTTGIRToTLX.cpp @@ -126,7 +126,7 @@ static const TTGIRToTLXMapping opMappings[] = { {"ttg.memdesc_reinterpret", "tlx.local_reinterpret", "Reinterpret buffer dtype/shape"}, {"ttng.tmem_subslice", "tlx.subslice", "TMEM subslice (Blackwell)"}, - {"ttg.memdesc_index", "tlx.memdesc_index", "Index into memdesc"}, + {"ttg.memdesc_index", "tlx.local_view", "Index into memdesc (buffer view)"}, // Async copy operations (cp.async) {"ttg.async_load", "tlx.async_load",