Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[DOC] Questions on Integrating a New CPU Operator into TorchAO? #1699

Open
Zijie-Tian opened this issue Feb 11, 2025 · 1 comment
Open
Labels
cpu question Further information is requested

Comments

@Zijie-Tian
Copy link

Zijie-Tian commented Feb 11, 2025

I'm working on integrating a CPU operator into TorchAO and have a few questions regarding the process:

How can I add a New CPU Operator in 'torchao/csrc':

  • What is the recommended approach for adding a new CPU operator in the 'csrc' directory?

  • Are there any specific guidelines or templates I should follow to ensure compatibility with the existing codebase?

How can I Remove or Disable current CUDA Operators:

  • How can I remove or disable all existing CUDA operators in the codebase?

  • Are there any configuration flags or build options that can be used to exclude CUDA-related code during compilation?

How can I Move Experimental MPS and CPU Code to TorchAO:

  • I noticed that there is experimental code for MPS and CPU in the repository (torchao/experimental/kernels'). What is the process for moving this code into the main TorchAO module?

  • Are there any specific considerations or steps I should follow to ensure a smooth transition?

Thank you for your help!

@supriyar
Copy link
Contributor

@metascroy can you help answer these since you have the most context?

@supriyar supriyar added question Further information is requested cpu labels Feb 12, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cpu question Further information is requested
Projects
None yet
Development

No branches or pull requests

2 participants