Currently, we re-initialize the environment for Neuron when torch-neuronx is present. In this change, we decouple the initialization when torch-neuronx is present. In the meantime, we also short circuit the returning value if libneuronxla is not present.
Currently, we re-initialize the environment for Neuron when
torch-neuronx
is present. In this change, we decouple the initialization whentorch-neuronx
is present. In the meantime, we also short circuit the returning value iflibneuronxla
is not present.