Skip to content

Tags: jeffhataws/xla

Tags

v2.1.1

Toggle v2.1.1's commit message
XLA bindings for torch_neuronx.trace and profile bindings (pytorch#5502)

* Add python bindings to set IR debug settings on and off (in addition to env variables)

* Add python bindings for turning device cache on and off, and special scalar handling

* Lint fixes

* Lint fixes for init_python_bindings.cpp