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

Support CUDA for conv1d, conv2d #132

Closed
wants to merge 2 commits into from
Closed

Support CUDA for conv1d, conv2d #132

wants to merge 2 commits into from

Conversation

jeffreysijuntan
Copy link
Contributor

Summary: Support CUDA for conv1d and conv2d

Differential Revision: D21885098

jeffreysijuntan added 2 commits June 4, 2020 10:17
Summary:
This diff does the two following thing:

1. Implement a encoder for LongTensor that embed fixed-point arithmetic into floating-point arithmetic. It enables us to utilize existing CUDA kernel for floating points to compute matrix multiplication for LongTensor.

2.  Add a new test file test_cuda.py for future CUDA related testing

The encoder currently only supports matmul. Conv2d will be supported in future diffs.

The API design is based on: D21860688

Differential Revision: D21882865

fbshipit-source-id: b69a0aa333f4d2f169dba30bcc900fd1ae4db224
Summary: Support CUDA for conv1d and conv2d

Differential Revision: D21885098

fbshipit-source-id: 1617e4dc78ee5c95989cad1bfc64e08f60733ede
@facebook-github-bot facebook-github-bot added CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported labels Jun 4, 2020
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D21885098

@knottb knottb closed this Jun 30, 2020
tanjuntao pushed a commit to tanjuntao/CrypTen that referenced this pull request Nov 27, 2023
Summary:
Pull Request resolved: fairinternal/CrypTen#132

also change rst to specify language of code block

(Note: this ignores all push blocking failures!)

Reviewed By: marksibrahim

Differential Revision: D17844355

fbshipit-source-id: 34abcdd2a179cc5fea362170a1be9fe58d2b9db5
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants