Cherry-pick 2.1 release branch into `xrt` through 9/18 (#5607)
* Handle dynamo function without input (#5565) (#5577)
* Make cpu tensor on XLA dynamo backend a warning instead of error (#5549) (#5576)
* [author: jluntamazon] Adding more explicit HLO lowering control by exposing LoweringContext… (#5431) (#5580)
* Adding more explicit HLO lowering control by exposing LoweringContext (and utilities) to python for Neuron
* fixing linter issues
* fixing spacing
* apply comments and fix compilation errors
* add test for new apis
* fix linter
* update test
* update test
* modify test
* reverse back to GetIrValue()
* update test inputs with random numbers
* skip unittest because it only fails in CI
---------
Co-authored-by: aws-kingrj <78175353+aws-kingrj@users.noreply.github.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-3-186.us-west-2.compute.internal>
Co-authored-by: seanlatias <seanlatias@gmail.com>
* fixing num_local_processes typo (#5573) (#5579)
Co-authored-by: aws-kingrj <78175353+aws-kingrj@users.noreply.github.com>
* Move where clear pending IR is called to avoid crash (#5552) (#5582)
* Move where clear pending IR is called to avoid crash
* fix CI
* fix CI and add some debugging messages
* Fix release branch and tag patterns for GitHub Actions (#5587) (#5590)
* Improve bernoulli rng-bit-generation memory footprint (#5581) (#5589)
* Allow downcasting RngUniform genenration for Bernoulli
Co-authored-by: Yeounoh Chung <yeounoh@google.com>
* Enable xla:gpu autocast for bfloat16 if not restricted (#5570) (#5591)
* Enable autocast for XLA:GPU
* linter fix
* XLA autocast test for GPU and TPU
* linter fix
* Ensure that xla autocast is properly enabled for GPU and does not crash when torch cuda is not available.
* linter fix
* Add tests
* Support bf16
* linter fix
* exclude unsupported test cases
* increase GPU test timeout to 300
Co-authored-by: Yeounoh Chung <yeounoh@google.com>
* Cherry-pick: Don't trigger CI build on release tag push (#5595)
Copy of #5594 on release branch
* formatting
---------
Co-authored-by: JackCaoG <59073027+JackCaoG@users.noreply.github.com>
Co-authored-by: Wonjoo Lee <wonjoo@google.com>
Co-authored-by: aws-kingrj <78175353+aws-kingrj@users.noreply.github.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-3-186.us-west-2.compute.internal>
Co-authored-by: seanlatias <seanlatias@gmail.com>
Co-authored-by: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Co-authored-by: Yeounoh Chung <yeounoh@google.com>
