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.

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

Fix a unit test in the PyTorch wrapper #663

Merged
merged 3 commits into from
May 12, 2022

Commits on May 11, 2022

  1. Fix a unit test in the PyTorch wrapper

    This test checked whether the allocator was set to the PyTorch allocator
    when the PyTorch shim is used. However, this is not the case when
    PyTorch is installed, but CuPy isn't, so the test would fail. Since this
    test relies on CuPy, disable it when CuPy is not available.
    danieldk committed May 11, 2022
    Configuration menu
    Copy the full SHA
    b262472 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    21610de View commit details
    Browse the repository at this point in the history
  3. Fix merge fallout

    danieldk committed May 11, 2022
    Configuration menu
    Copy the full SHA
    5a63643 View commit details
    Browse the repository at this point in the history