search

microsoft / knossos-ksc

Compiler with automatic differentiation
Other
45 stars 10 forks source link

Use TORCH_EXTENSION_NAME macro in generated C++ #981

Closed dcrc2 closed 3 years ago

dcrc2 commented 3 years ago

When compiling a PyTorch extension, use the TORCH_EXTENSION_NAME macro in the generated C++, to avoid the problem described at https://github.com/microsoft/knossos-ksc/pull/970#issuecomment-887429874