Int4 Cute Dsl Repack#
- int32_t trt_edgellm::kernel::int4CuteDslFragmentRows(
- int32_t N,
- int32_t K
Number of INT8
[rows, 512]rows required by Int4GroupwiseGemmPluginV2.
- cudaError_t trt_edgellm::kernel::launchModelOptInt4CuteDslRepack(
- uint8_t const *weightNhalfK,
- int8_t *fragmentRows512,
- int32_t N,
- int32_t K,
- cudaStream_t stream
Repack ModelOpt packed
[N/2,K]weights directly from checkpoint storage.
- cudaError_t trt_edgellm::kernel::launchGptqInt4CuteDslRepack(
- int32_t const *qweightK8N,
- int32_t const *qzerosGN8,
- int32_t const *activationPermutation,
- int8_t *fragmentRows512,
- int32_t N,
- int32_t K,
- int32_t numGroups,
- int32_t groupSize,
- int32_t zeroPointOffset,
- cudaStream_t stream
Repack GPTQ
[K/8,N]weights directly from checkpoint storage.
- cudaError_t trt_edgellm::kernel::launchAwqInt4CuteDslRepack(
- int32_t const *qweightKN8,
- int32_t const *qzerosGN8,
- int8_t *fragmentRows512,
- int32_t N,
- int32_t K,
- int32_t numGroups,
- int32_t groupSize,
- cudaStream_t stream
Repack classic AWQ
[K,N/8]weights directly from checkpoint storage.