Support cuTENSOR - Githubissues

conda-forge / cupy-feedstock

A conda-smithy repository for cupy.

BSD 3-Clause "New" or "Revised" License

5 stars 23 forks source link

Support cuTENSOR #94

Closed leofang closed 3 years ago

leofang commented 3 years ago

Checklist

[x] Used a personal fork of the feedstock to propose changes
[x] Bumped the build number (if the version is unchanged)
[ ] Reset the build number to 0 (if the version changed)
[x] Re-rendered with the latest conda-smithy (Use the phrase code>@<space/conda-forge-admin, please rerender in a comment in this PR for automated rerendering)
[x] Ensured the license file is being packaged.

conda-forge-linter commented 3 years ago

Hi! This is the friendly automated conda-forge-linting service.

I just wanted to let you know that I linted all conda-recipes in your PR (recipe) and found it was in an excellent condition.

leofang commented 3 years ago

@conda-forge-admin, please rerender

leofang commented 3 years ago

@conda-forge-admin, please rerender

github-actions[bot] commented 3 years ago

Hi! This is the friendly automated conda-forge-webservice. I tried to rerender for you, but it looks like there was nothing to do.

leofang commented 3 years ago

@conda-forge/cupy This PR expands the build matrix to support cuTENSOR. All tests passed, PTAL.

cuTENSOR works with CUDA 10.1, 10.2, and 11+. On CUDA 11.0, it works straightforwardly as the glibc version 2.17 used in the CentOS 7 image is new enough. On 10.1/10.2, however, the CentOS 6 image has glibc 2.12, which cuTENSOR does not support, so in additional to the original builds I also add the combination of glibc 2.17 + CUDA 10.1/10.2 to enable cuTENSOR, which adds an extra 12 outputs. If this strategy is OK I'll port it to the rc branch (#95).

leofang commented 3 years ago

cc: @jakirkham

leofang commented 3 years ago

Merge to prepare for the upcoming release.

jakirkham commented 3 years ago

Thanks Leo! 😄