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 226 days ago

xin3he requested a review from

wenhuach21 226 days ago

xin3he requested a review from

n1ck-guo 226 days ago

fix device mismatch issue

4d66376c

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

unfiy changes for qkv and mlp

a8d8d6c3

chensuyue added this to the 0.9.3 milestone 225 days ago

add ut

4ac3dd14

xin3he force pushed from aeea5c27 to 4ac3dd14 225 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 225 days ago

add UT to check enable_torch_compile

9467ff14

wenhuach21 approved these changes on 2025-12-23

chensuyue merged 13ba747f into main 224 days ago

chensuyue deleted the xinhe/bug branch 224 days ago

Reviewers

wenhuach21

WeiweiZhang1

n1ck-guo

Assignees

No one assigned

Labels

None yet

Milestone

0.9.3