Skip to content

Conversation

@metascroy
Copy link
Contributor

This enables you to run the following on macOS:

python setup.py develop

Python extensions always look for so files with specific names.

@pytorch-bot
Copy link

pytorch-bot bot commented Aug 12, 2025

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/2748

Note: Links to docs will display an error until the docs builds have been completed.

⏳ No Failures, 2 Pending

As of commit d694f32 with merge base 758f744 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Aug 12, 2025
@metascroy metascroy requested a review from swolchok August 12, 2025 18:10
@metascroy metascroy added the topic: not user facing Use this tag if you don't want this PR to show up in release notes label Aug 12, 2025
@metascroy metascroy requested a review from drisspg August 12, 2025 18:11
Copy link

@swolchok swolchok left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Python extensions always look for so files with specific names.

I was skeptical of this, but sure enough .so is the more popular extension in my site-packages directory. No complaints from my end but would like to give @drisspg a chance to review.



def find_and_load_libtorchao_ops(potential_paths):
# import torchao._experimental_aten_ops
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: maybe a doc string?

Copy link
Contributor

@drisspg drisspg left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think this is fine, is there any doc on libs are searched and this name schema you mention?

@metascroy metascroy merged commit e6b38bb into main Aug 18, 2025
34 checks passed
@metascroy
Copy link
Contributor Author

I think this is fine, is there any doc on libs are searched and this name schema you mention?

Added docstring to op load function, but this is not user-facing. It is called inside torchao's init.py automatically.

liangel-02 pushed a commit that referenced this pull request Aug 25, 2025
* Fix setup.py develop workflow

* up

* up
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: not user facing Use this tag if you don't want this PR to show up in release notes

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants