Skip to content

Fix QKVCrossParallelLinear::sync_weight_attrs for PyTorch compile#1184

Merged
afierka-intel merged 1 commit intohabana_mainfrom
dev/anko/modal_dict_fix
May 16, 2025
Merged

Fix QKVCrossParallelLinear::sync_weight_attrs for PyTorch compile#1184
afierka-intel merged 1 commit intohabana_mainfrom
dev/anko/modal_dict_fix

Conversation

@anko-intel
Copy link
Copy Markdown

@anko-intel anko-intel commented Apr 29, 2025

It fixes issue with compilation in Dynamo. Original version failed with:

torch._dynamo.exc.InternalTorchDynamoError: AttributeError: 'dict' object has no attribute 'node'
...
    from user code:
..
         File "./vllm/model_executor/layers/linear.py", line 1469, in sync_weight_attrs
           for k in (set(src_param.__dict__.keys()) -

@anko-intel
Copy link
Copy Markdown
Author

/run-gaudi-tests

Comment thread vllm/model_executor/layers/linear.py
@anko-intel
Copy link
Copy Markdown
Author

/run-gaudi-tests

1 similar comment
@anko-intel
Copy link
Copy Markdown
Author

/run-gaudi-tests

@michalkuligowski
Copy link
Copy Markdown

@kzawora-intel @madamczyk-intel I think we could merge it since we have PR on upstream?

@anko-intel
Copy link
Copy Markdown
Author

/run-gaudi-tests

@anko-intel
Copy link
Copy Markdown
Author

@michalkuligowski, upstream version vllm-project#17844 was merged.

@anko-intel anko-intel force-pushed the dev/anko/modal_dict_fix branch from df53121 to 951eb97 Compare May 15, 2025 08:38
@anko-intel
Copy link
Copy Markdown
Author

/run-gaudi-tests

@afierka-intel afierka-intel merged commit 387b93a into habana_main May 16, 2025
46 checks passed
@afierka-intel afierka-intel deleted the dev/anko/modal_dict_fix branch May 16, 2025 10:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants