Add some extra flags to TorchModule.compile and remove GOOGLE_CUDA from Template
Showing
- src/pystencils_autodiff/backends/astnodes.py 7 additions, 5 deletionssrc/pystencils_autodiff/backends/astnodes.py
- src/pystencils_autodiff/backends/module.tmpl.cpp 4 additions, 4 deletionssrc/pystencils_autodiff/backends/module.tmpl.cpp
- src/pystencils_autodiff/framework_integration/astnodes.py 6 additions, 2 deletionssrc/pystencils_autodiff/framework_integration/astnodes.py
Please register or sign in to comment