Uploaded image for project: 'AI Platform Core Components'
  1. AI Platform Core Components
  2. AIPCC-6267

Build torch xla from source instead of using pre-built wheel

    • Icon: Bug Bug
    • Resolution: Unresolved
    • Icon: Minor Minor
    • None
    • None
    • Accelerator Enablement
    • None
    • False
    • Hide

      None

      Show
      None
    • False
    • AIPCC Accelerators 18, AIPCC Accelerators 19, AIPCC Accelerators 20

      Description of problem:

          Currently, torch xla (dependency of torch on tpu) is used as a pre-built variant for tpu. While on PyPI they don't publish an sdist, the source code is available at https://github.com/pytorch/xla/tags. Thus, we don't have to use the pre-built wheel and can build them ourselves from sources.
      

              rh-ee-jelevy Jenna Levy
              unassigned_jira Unassigned
              Frank's Team
              Votes:
              0 Vote for this issue
              Watchers:
              2 Start watching this issue

                Created:
                Updated: