MCPcopy Create free account
hub / github.com/NVIDIA/cuda-tile / mlirCudaTileStridedViewTypeGet

Function mlirCudaTileStridedViewTypeGet

lib/CAPI/Dialect/CudaTileDialect.cpp:249–264  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

247}
248
249MlirType mlirCudaTileStridedViewTypeGet(
250 MlirContext ctx, MlirAttribute tileShapeAttr,
251 MlirAttribute traversalStridesAttr, MlirType tensorViewType,
252 intptr_t dimMapRank, const int32_t *dimMap, MlirAttribute paddingValue) {
253 ArrayRef<int32_t> dimMapRef(dimMap, dimMapRank);
254 auto tileShapeDenseI32 = cast<DenseI32ArrayAttr>(unwrap(tileShapeAttr));
255 auto traversalStridesDenseI32 =
256 cast<DenseI32ArrayAttr>(unwrap(traversalStridesAttr));
257 auto tensorView = cast<TensorViewType>(unwrap(tensorViewType));
258 PaddingValueAttr paddingValueAttr = nullptr;
259 if (!mlirAttributeIsNull(paddingValue))
260 paddingValueAttr = cast<PaddingValueAttr>(unwrap(paddingValue));
261 return wrap(StridedViewType::get(unwrap(ctx), tileShapeDenseI32,
262 traversalStridesDenseI32, tensorView,
263 dimMapRef, paddingValueAttr));
264}
265
266MlirAttribute mlirCudaTileStridedViewTypeGetTileShape(MlirType type) {
267 return wrap(cast<StridedViewType>(unwrap(type)).getTileShape());

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected