How to automatically set the backends? about extension-cpp HOT 2 CLOSED

Also, can we merge the two setup.py together and make them share the same package name?
Sorry to close this issue mistakenly ...

from extension-cpp.

You can merge the two Functions and write:

if tensor.is_cuda():
  # call CUDA code
else:
  # call C++ code

inside your PyTorch Function, or you can move it into C++ and write:

if (tensor.is_cuda()) {
  // call CUDA code
} else {
  // call C++ code
}

There's nothing to stop you from merging the setup.pys, no. I won't do it for this example because things are supposed to be isolated.

from extension-cpp.