Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions src/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -187,6 +187,7 @@ target_link_libraries(flm PUBLIC
phi4_npu
nanbeige_npu
dequant
dequant_new
gemm
lm_head
mha
Expand Down
17 changes: 11 additions & 6 deletions src/include/modules/dequant.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -21,12 +21,17 @@ class Dequant{
Dequant(LM_Config& config);
~Dequant();

/// @brief generate the dequant sequence
/// @param seq: the sequence
/// @param D_in: input dimension of the projection weight
/// @param D_out: output dimension of the projection weight
/// @param weight_offset: the weight offset in byte
/// @param mode: dequant output mode
typedef enum: int {
Q4_1 = 0,
Q8_0 = 1
} quant_block_t;

void reorder_cpy(u8 *dst, buffer<u8> &src,
quant_block_t quant_block_type,
const int quant_matrix_row,
const int quant_matrix_col,
const int vertical_blocks=2,
const int vetrical_block_interleave_byte_size=-1);
void generate_dequant_q4_1_seq(npu_sequence* seq, const uint32_t D_in, const uint32_t D_out, const uint32_t weight_offset, int mode);
void generate_dequant_q80_packed_in_q4nx_seq(npu_sequence* seq, const uint32_t D_in, const uint32_t D_out, const uint32_t weight_offset, int mode);
private:
Expand Down
Binary file added src/lib/libdequant_new.so
Binary file not shown.
Binary file modified src/lib/libllama_npu.so
Binary file not shown.
Binary file modified src/lib/libphi4_npu.so
Binary file not shown.
Binary file modified src/lib/libq4_npu_eXpress.so
Binary file not shown.
Binary file modified src/lib/libqwen3_5vl_npu.so
Binary file not shown.
Binary file modified src/lib/libqwen3_npu.so
Binary file not shown.
Binary file modified src/lib/libqwen3vl_npu.so
Binary file not shown.
1 change: 1 addition & 0 deletions src/test/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -110,6 +110,7 @@ function(add_npu_test TARGET_NAME OUTPUT_SUBDIR)
q4_npu_eXpress
lm_head
dequant
dequant_new
gemm
mha
tokenizers_cpp
Expand Down
Binary file added src/xclbins/Qwen3.5-0.8B-NPU2/dequant_q41.xclbin
Binary file not shown.
Binary file added src/xclbins/Qwen3.5-0.8B-NPU2/dequant_q80.xclbin
Binary file not shown.
Binary file modified src/xclbins/Qwen3.5-0.8B-NPU2/layer.xclbin
Binary file not shown.
Binary file modified src/xclbins/Qwen3.5-0.8B-NPU2/lm_head.xclbin
Binary file not shown.