xla
547ea8b5 - XLA bindings for torch_neuronx.trace and profile bindings (#5502)

Commit
2 years ago
XLA bindings for torch_neuronx.trace and profile bindings (#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
Author
Parents
Loading