Skip to content

Commit

Permalink
Choose cupy dlpack method based on version (#10631)
Browse files Browse the repository at this point in the history
This PR adds a CuPy version check to the 10 minute CuPy notebook, using this to decide which DLPack input method to use - this should allow us to support older CuPy versions that don't yet have `from_dlpack` while also supporting newer versions which will eventually deprecate `fromDlpack`.

Closes #10612

Authors:
   - Charles Blackmon-Luca (https://github.com/charlesbluca)

Approvers:
   - GALI PREM SAGAR (https://github.com/galipremsagar)
  • Loading branch information
charlesbluca authored Apr 11, 2022
1 parent b50ae82 commit 0cb75a4
Showing 1 changed file with 123 additions and 124 deletions.
Loading

0 comments on commit 0cb75a4

Please sign in to comment.