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

Introduce CUDA OpenXLA fallback. #7318

Merged
merged 27 commits into from
Jul 3, 2024
Merged

Introduce CUDA OpenXLA fallback. #7318

merged 27 commits into from
Jul 3, 2024

Commits on Jul 1, 2024

  1. Initial implementation.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    091fb6b View commit details
    Browse the repository at this point in the history

  2. Replace usage of xla_cpu_fallback.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    1c8b479 View commit details
    Browse the repository at this point in the history

  3. Remove 'fallback' suffix.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    a908ed4 View commit details
    Browse the repository at this point in the history

  4. Scratch

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    2d4bb59 View commit details
    Browse the repository at this point in the history

  5. Deal with undefined tensors + bring back step 3.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    aae7197 View commit details
    Browse the repository at this point in the history

  6. Refactor test.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    bbb86a8 View commit details
    Browse the repository at this point in the history

  7. Refactor fallback.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    540275a View commit details
    Browse the repository at this point in the history

  8. Refactor tests.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    868333a View commit details
    Browse the repository at this point in the history

  9. Fix lint issues.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    e9fb89d View commit details
    Browse the repository at this point in the history

  10. Forward declare CUDA functions.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    e1e1da1 View commit details
    Browse the repository at this point in the history

  11. Lazy initialize CUDA.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    5413a74 View commit details
    Browse the repository at this point in the history

  12. Fix lint.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    1304dc7 View commit details
    Browse the repository at this point in the history

  13. Fix compilation.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    e4642f9 View commit details
    Browse the repository at this point in the history

  14. Address reviews.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    2f32ba8 View commit details
    Browse the repository at this point in the history

  15. Create fallback implementation for tests.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    8c0dddd View commit details
    Browse the repository at this point in the history

  16. Fix lint issue.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    9c2e28b View commit details
    Browse the repository at this point in the history

  17. Fix test builds and add libc10_cuda.so dependency.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    ea33a74 View commit details
    Browse the repository at this point in the history

  18. Fix C++ test dependencies.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    0ef4e1b View commit details
    Browse the repository at this point in the history

  19. Build a second library for conditionally loading CUDA functions.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    d035935 View commit details
    Browse the repository at this point in the history

  20. Add Python module initialization.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    4129e86 View commit details
    Browse the repository at this point in the history

  21. Fix lib and dlopen options.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    9a22be7 View commit details
    Browse the repository at this point in the history

  22. Fix lint issues.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    7d5063a View commit details
    Browse the repository at this point in the history

  23. Skip CUDA fallback tests on CPU envs.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    75ffb01 View commit details
    Browse the repository at this point in the history

  24. Fix lint issues.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    0dd29da View commit details
    Browse the repository at this point in the history

  25. Clean up.

    @ysiraichi
    ysiraichi committed Jul 1, 2024
    Configuration menu
    Copy the full SHA
    2c2b165 View commit details
    Browse the repository at this point in the history

Commits on Jul 2, 2024

  1. Make sure only one XLA device is used.

    @ysiraichi
    ysiraichi committed Jul 2, 2024
    Configuration menu
    Copy the full SHA
    4686238 View commit details
    Browse the repository at this point in the history

  2. Fix compilation issues.

    @ysiraichi
    ysiraichi committed Jul 2, 2024
    Configuration menu
    Copy the full SHA
    8d5fe8f View commit details
    Browse the repository at this point in the history