rename sharded_layout => blocked_layout

2022-06-05 16:14:59 +08:00
parent bbf75b492f
commit 7807f64ef3
7 changed files with 98 additions and 96 deletions
--- a/lib/Dialect/TritonGPU/Transforms/TritonGPUConversion.cpp
+++ b/lib/Dialect/TritonGPU/Transforms/TritonGPUConversion.cpp
@@ -46,7 +46,7 @@ TritonGPUTypeConverter::TritonGPUTypeConverter(MLIRContext *context,
      remainingThreads /= blockTileSize[dim];
      // TODO: will we need repetition?
    }
-    Attribute encoding = triton::gpu::TritonGPUShardedEncodingAttr::get(
+    Attribute encoding = triton::gpu::TritonGPUBlockedEncodingAttr::get(
        context, threadTileSize, warpTileSize, blockTileSize, order);
    return RankedTensorType::get(shape, elementType, encoding);
  });