Home
last modified time | relevance | path

Searched defs:BLayout_TC_int4 (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dint4mm.cu447 struct BLayout_TC_int4 { struct
448 static constexpr int32_t kInnerKTiles = InnerKTiles;
449 static constexpr int32_t kMTileSize = 16;
451 static constexpr int32_t kNTileSize = 16;
453 static constexpr int32_t kNTileSize = 8;
455 static constexpr int32_t kKTileSize = 16;
458 static __device__ void load( in load()