There was an error fetching the commit references. Please try again later.
Allow calling Torch kernels with dynamic shape
Showing
- src/pystencils_autodiff/_autodiff.py 1 addition, 1 deletionsrc/pystencils_autodiff/_autodiff.py
- src/pystencils_autodiff/backends/_torch_native.py 11 additions, 5 deletionssrc/pystencils_autodiff/backends/_torch_native.py
- src/pystencils_autodiff/framework_integration/astnodes.py 89 additions, 81 deletionssrc/pystencils_autodiff/framework_integration/astnodes.py
Please register or sign in to comment