You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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!
The text was updated successfully, but these errors were encountered:
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!
The text was updated successfully, but these errors were encountered: