Chih-Chieh Yang b690e34824 [Model] Mamba2 preallocate SSM output tensor to avoid d2d copy overhead (#21075) ...

Signed-off-by: Chih-Chieh Yang <7364402+cyang49@users.noreply.github.com>
Signed-off-by: Chih-Chieh-Yang <7364402+cyang49@users.noreply.github.com>

2025-08-02 01:59:34 -07:00

..

layers

[Model] Mamba2 preallocate SSM output tensor to avoid d2d copy overhead (#21075)

2025-08-02 01:59:34 -07:00

model_loader

[Quantization] Enable BNB support for InternS1 (#21953)

2025-08-01 11:09:54 +00:00

models

[Model] Mamba2 preallocate SSM output tensor to avoid d2d copy overhead (#21075)

2025-08-02 01:59:34 -07:00

__init__.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

custom_op.py

[V0 deprecation] Remove V0 HPU backend (#21131)

2025-07-17 16:37:36 -07:00

parameter.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

pooling_metadata.py

[Model][1/N] Support multiple poolers at model level (#21227)

2025-07-21 02:22:21 -07:00

sampling_metadata.py

Revert "Update sampling_metadata.py (#21937)" (#22088)

2025-08-01 05:24:46 -07:00

utils.py

[Quantization] Enable BNB support for InternS1 (#21953)

2025-08-01 11:09:54 +00:00