e3nn / e3nn / 12209514216

76%

push

github

Force `CodeGenMixin` to entirely respect `e3nn.set_optimization_defaults` (#484)

* first draft of fix

* avoid creating dummy empty tensor for broadcasting that breaks tracing

* Remove error from fx GraphModule so it doesn't assert False on re-tracing at unpickle time

* shortcut for zero irreps

* cleanup

* Update e3nn/o3/_tensor_product/_codegen.py

Co-authored-by: Alby M. <1473644+Linux-cpp-lisp@users.noreply.github.com>

* Update e3nn/o3/_tensor_product/_codegen.py

Co-authored-by: Alby M. <1473644+Linux-cpp-lisp@users.noreply.github.com>

* broadcasting [:1] in codegen_right was causing issues

---------

Co-authored-by: Mit Kotak <mitkotak0305@gmail.com>
Co-authored-by: Mit Kotak <53411468+mitkotak@users.noreply.github.com>

23 of 28 new or added lines in 3 files covered. (82.14%)

3427 of 4456 relevant lines covered (76.91%)

0.77 hits per line