MCPcopy Create free account
hub / github.com/NVIDIA/cutlass / zipped_divide

Function zipped_divide

include/cute/tensor_impl.hpp:935–942  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

933template <class Tensor, class Tiler,
934 __CUTE_REQUIRES(is_tensor<remove_cvref_t<Tensor>>::value)>
935CUTE_HOST_DEVICE constexpr
936auto
937zipped_divide(Tensor && tensor,
938 Tiler const& tiler) // Layout or Tile<Layout...> or Shape
939{
940 return make_tensor(static_cast<Tensor&&>(tensor).data(),
941 zipped_divide(tensor.layout(), tiler));
942}
943
944// tiled_divide is zipped_divide with the second output mode flattened ((BLK_A,BLK_B),a,b,x,y)
945template <class Tensor, class Tiler,

Callers 2

inner_partitionFunction · 0.70
outer_partitionFunction · 0.70

Calls 3

make_tensorFunction · 0.70
dataMethod · 0.45
layoutMethod · 0.45

Tested by

no test coverage detected