Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Dockerfile.jax: install JAX for TE build #997

Merged
merged 17 commits into from
Aug 19, 2024
Merged

Dockerfile.jax: install JAX for TE build #997

merged 17 commits into from
Aug 19, 2024

Commits on Aug 19, 2024

  1. Dockerfile.jax: install JAX for TE build

    @olupton
    olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    524732d View commit details
    Browse the repository at this point in the history

  2. fixups

    @olupton
    olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    2b21119 View commit details
    Browse the repository at this point in the history

  3. Use Phuong branch for TE fix

    @DwarKapex @olupton
    DwarKapex authored and olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    6adfe3f View commit details
    Browse the repository at this point in the history

  4. Use Phuong branch for TE fix

    @DwarKapex @olupton
    DwarKapex authored and olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    b7239c1 View commit details
    Browse the repository at this point in the history

  5. Use Phuong branch for TE fix

    @DwarKapex @olupton
    DwarKapex authored and olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    f13d7e7 View commit details
    Browse the repository at this point in the history

  6. Fix for JAX build via bazel

    @DwarKapex @olupton
    DwarKapex authored and olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    6c2e52f View commit details
    Browse the repository at this point in the history

  7. No bazel cache

    @DwarKapex @olupton
    DwarKapex authored and olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    abb6d97 View commit details
    Browse the repository at this point in the history

  8. No bazel cache

    @DwarKapex @olupton
    DwarKapex authored and olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    3ea2046 View commit details
    Browse the repository at this point in the history

  9. Update repo env for bazel build

    @DwarKapex @olupton
    DwarKapex authored and olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    52440ef View commit details
    Browse the repository at this point in the history

  10. Symlink CUDA lib->lib64

    @DwarKapex @olupton
    DwarKapex authored and olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    fe88054 View commit details
    Browse the repository at this point in the history

  11. Set CUDANN version manually

    @DwarKapex @olupton
    DwarKapex authored and olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    791cb3a View commit details
    Browse the repository at this point in the history

  12. reduce diff

    @olupton
    olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    7a6e142 View commit details
    Browse the repository at this point in the history

  13. Use local NCCL/cuDNN

    @olupton
    olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    cff361b View commit details
    Browse the repository at this point in the history

  14. add new .bazelrc for local cuda/cudnn/nccl paths

    @olupton
    olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    1aa66ec View commit details
    Browse the repository at this point in the history

  15. fixup

    @olupton
    olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    88d5010 View commit details
    Browse the repository at this point in the history

  16. fixup arm

    @olupton
    olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    249fb7c View commit details
    Browse the repository at this point in the history

  17. Drop -links

    @olupton
    olupton committed Aug 19, 2024
    Configuration menu
    Copy the full SHA
    952ceaf View commit details
    Browse the repository at this point in the history