xla
41b38f55 - [author: jluntamazon] Adding more explicit HLO lowering control by exposing LoweringContext… (#5431)

Comment changes are shownComment changes are hidden
Commit
1 year ago
[author: jluntamazon] Adding more explicit HLO lowering control by exposing LoweringContext… (#5431) * 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: Ubuntu <ubuntu@ip-172-31-3-186.us-west-2.compute.internal> Co-authored-by: seanlatias <seanlatias@gmail.com>
Author
Parents
  • test
    • File
      test_operations.py
  • torch_xla/csrc
    • File
      init_python_bindings.cpp