PR #1175 fix device mismatch issue for unifying qkv scale

fix device mismatch issue for unifying qkv scale #1175

chensuyue merged 5 commits into main from xinhe/bug

xin3he requested a review from

WeiweiZhang1 7 days ago

xin3he requested a review from

wenhuach21 7 days ago

xin3he requested a review from

n1ck-guo 7 days ago

fix device mismatch issue

4d66376c

xin3he changed the title ~~fix device mismatch issue~~ fix device mismatch issue for unifying qkv scale 7 days ago

unfiy changes for qkv and mlp

a8d8d6c3

chensuyue added this to the 0.9.3 milestone 6 days ago

add ut

4ac3dd14

xin3he force pushed from aeea5c27 to 4ac3dd14 6 days ago

reset enable_torch_compile to `False` as nvfp4 is enabled (#1176)

5b9e7570

wenhuach21 commented on 2025-12-22

xin3he requested a review from

wenhuach21 6 days ago

add UT to check enable_torch_compile

9467ff14

wenhuach21 approved these changes on 2025-12-23

chensuyue merged 13ba747f into main 5 days ago

chensuyue deleted the xinhe/bug branch 5 days ago

Reviewers

wenhuach21

WeiweiZhang1

n1ck-guo

Assignees

No one assigned

Labels

None yet

Milestone

0.9.3