Add TAO dependency to cuda predictor #3082

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

Merged

facebook-github-bot merged 1 commit into pytorch:main from RandySheriff:export-D83214553

Sep 29, 2025

Contributor

RandySheriff commented Sep 26, 2025

Summary: In TorchAO, define a new sparse cutlass ops library to be linked to the predictor. This is to enable sparsity feature in coming HIM models.

Reviewed By: pranavsharma

Differential Revision: D83214553

pytorch-bot bot commented Sep 26, 2025 •

edited

Loading

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3082

📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 660ae46 with merge base de92bdc ():
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

meta-cla bot added the CLA Signed label

Contributor

facebook-github-bot commented Sep 26, 2025

@RandySheriff has exported this pull request. If you are a Meta employee, you can view the originating diff in D83214553.

facebook-github-bot added fb-exported meta-exported labels

RandySheriff added a commit to RandySheriff/ao that referenced this pull request


          Add TAO dependency to cuda predictor (pytorch#3082)

8d2696f

Summary:

In TorchAO, define a new sparse cutlass ops library to be linked to the predictor. This is to enable sparsity feature in coming HIM models.

Reviewed By: pranavsharma

Differential Revision: D83214553

RandySheriff force-pushed the export-D83214553 branch from 8f22bf2 to 8d2696f Compare

September 26, 2025 18:06

Contributor

facebook-github-bot commented Sep 26, 2025

@RandySheriff has exported this pull request. If you are a Meta employee, you can view the originating diff in D83214553.

RandySheriff force-pushed the export-D83214553 branch from 8d2696f to 6785af4 Compare

September 26, 2025 18:14

RandySheriff added a commit to RandySheriff/ao that referenced this pull request


          Add TAO dependency to cuda predictor (pytorch#3082)

6785af4

Summary:

In TorchAO, define a new sparse cutlass ops library to be linked to the predictor. This is to enable sparsity feature in coming HIM models.

Reviewed By: pranavsharma

Differential Revision: D83214553

Contributor

facebook-github-bot commented Sep 26, 2025

@RandySheriff has exported this pull request. If you are a Meta employee, you can view the originating diff in D83214553.

jcaip added sparsity topic: not user facing labels

RandySheriff force-pushed the export-D83214553 branch from 6785af4 to e1169c5 Compare

September 26, 2025 20:00

RandySheriff added a commit to RandySheriff/ao that referenced this pull request


          Add TAO dependency to cuda predictor (pytorch#3082)

e1169c5

Summary:

In TorchAO, define a new sparse cutlass ops library to be linked to the predictor. This is to enable sparsity feature in coming HIM models.

Reviewed By: pranavsharma

Differential Revision: D83214553

Contributor

facebook-github-bot commented Sep 26, 2025

@RandySheriff has exported this pull request. If you are a Meta employee, you can view the originating diff in D83214553.

RandySheriff added a commit to RandySheriff/ao that referenced this pull request


          Add TAO dependency to cuda predictor (pytorch#3082)

c83765c

Summary:

In TorchAO, define a new sparse cutlass ops library to be linked to the predictor. This is to enable sparsity feature in coming HIM models.

Reviewed By: pranavsharma

Differential Revision: D83214553

RandySheriff force-pushed the export-D83214553 branch from e1169c5 to c83765c Compare

September 26, 2025 21:54

Contributor

facebook-github-bot commented Sep 26, 2025

@RandySheriff has exported this pull request. If you are a Meta employee, you can view the originating diff in D83214553.

RandySheriff force-pushed the export-D83214553 branch from c83765c to 04a43e3 Compare

September 27, 2025 04:29

RandySheriff added a commit to RandySheriff/ao that referenced this pull request


          Add TAO dependency to cuda predictor (pytorch#3082)

04a43e3

Summary:

In TorchAO, define a new sparse cutlass ops library to be linked to the predictor. This is to enable sparsity feature in coming HIM models.

Reviewed By: pranavsharma

Differential Revision: D83214553

Contributor

facebook-github-bot commented Sep 27, 2025

@RandySheriff has exported this pull request. If you are a Meta employee, you can view the originating diff in D83214553.

RandySheriff added a commit to RandySheriff/ao that referenced this pull request


          Define a TorchAO cutlass sparse op lib (pytorch#3082)

b326ce1

Summary:
As titled ~ this is to be linked to predictor by next diff.

Reviewed By: pranavsharma

Differential Revision: D83214553

RandySheriff force-pushed the export-D83214553 branch from 04a43e3 to b326ce1 Compare

September 27, 2025 19:48

Contributor

facebook-github-bot commented Sep 27, 2025

@RandySheriff has exported this pull request. If you are a Meta employee, you can view the originating diff in D83214553.

RandySheriff added a commit to RandySheriff/ao that referenced this pull request


          Define a TorchAO cutlass sparse op lib (pytorch#3082)

68aa23f

Summary:
As titled ~ this is to be linked to predictor by next diff.

Reviewed By: pranavsharma

Differential Revision: D83214553

RandySheriff force-pushed the export-D83214553 branch from b326ce1 to 68aa23f Compare

September 27, 2025 22:10

Contributor

facebook-github-bot commented Sep 27, 2025

@RandySheriff has exported this pull request. If you are a Meta employee, you can view the originating diff in D83214553.


          Define a TorchAO cutlass sparse op lib (pytorch#3082)

660ae46

Summary:
As titled ~ this is to be linked to predictor by next diff.

Reviewed By: pranavsharma

Differential Revision: D83214553

RandySheriff force-pushed the export-D83214553 branch from 68aa23f to 660ae46 Compare

September 29, 2025 15:51

Contributor

facebook-github-bot commented Sep 29, 2025

@RandySheriff has exported this pull request. If you are a Meta employee, you can view the originating diff in D83214553.

jcaip approved these changes

View reviewed changes

Contributor

jcaip left a comment

lgtm

facebook-github-bot merged commit 2918c5e into pytorch:main

17 of 20 checks passed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed fb-exported meta-exported sparsity topic: not user facing