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

Build Tensorflow CUDA wheels from source using Bazel

    • Icon: Epic Epic
    • Resolution: Done
    • Icon: Undefined Undefined
    • None
    • None
    • Accelerator Enablement
    • None
    • Build Tensorflow wheels from source using Bazel
    • False
    • Hide

      None

      Show
      None
    • False
    • In Progress
    • AIPCC-7810 - Build TensorFlow from source in RHOAI
    • AIPCC-7810Build TensorFlow from source in RHOAI
    • 0% To Do, 0% In Progress, 100% Done
    • AIPCC Accelerators 21, AIPCC Accelerators 22, AIPCC Accelerators 23, AIPCC Accelerators 24, AIPCC Accelerators 25

      Why:

      To build TensorFlow from source, we need to be able to run Bazel from Fromager, in our AIPCC builder, it's the only way to build the wheel.

      What:

      We need to find a way to execute Bazel in our ecosystem among the dependencies to build TensorFlow. Both Bazel and the dependencies will be tracked via another EPIC.

      How:

      A prototype was proposed here: https://gitlab.com/redhat/rhel-ai/wheels/builder/-/merge_requests/1711

      Note: there is a remaining issue where we'd want to drop the "dev0" from the wheel name. Something was tried in the MR but it didn't work yet.

      Scope of this EPIC:

      This will track the work done in a Tensorflow package plugin and the addition of this package into the right collections for the right variants. Bazel and the dependencies will be figured out in different EPICs.

              spryor@redhat.com Sean Pryor
              emacchi@redhat.com Emilien Macchi
              Frank's Team
              Votes:
              0 Vote for this issue
              Watchers:
              1 Start watching this issue

                Created:
                Updated:
                Resolved: