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

Function mlirCudaTileTensorViewTypeGetRank

lib/CAPI/Dialect/CudaTileDialect.cpp:142–144  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

140}
141
142intptr_t mlirCudaTileTensorViewTypeGetRank(MlirType type) {
143 return cast<TensorViewType>(unwrap(type)).getShape().size();
144}
145
146int64_t mlirCudaTileTensorViewTypeGetDimSize(MlirType type, intptr_t pos) {
147 return cast<TensorViewType>(unwrap(type)).getShape()[pos];

Callers 1

NB_MODULEFunction · 0.85

Calls 1

sizeMethod · 0.80

Tested by

no test coverage detected