https://github.com/szagoruyko/openai-gemm.pytorch
PyTorch bindings for openai-gemm
https://github.com/szagoruyko/openai-gemm.pytorch
gemm pytorch
Last synced: 5 months ago
JSON representation
PyTorch bindings for openai-gemm
- Host: GitHub
- URL: https://github.com/szagoruyko/openai-gemm.pytorch
- Owner: szagoruyko
- Created: 2017-02-06T20:36:07.000Z (over 8 years ago)
- Default Branch: master
- Last Pushed: 2017-02-06T20:36:22.000Z (over 8 years ago)
- Last Synced: 2025-01-01T18:35:04.637Z (5 months ago)
- Topics: gemm, pytorch
- Language: Python
- Size: 1.95 KB
- Stars: 20
- Watchers: 4
- Forks: 4
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
openai-gemm.pytorch
========
PyTorch bindings for openai-gemm.
## Installation
Clone original openai-gemm and add it to PYTHONPATH,
install pycuda:
```
pip install pycuda
```
and follow instructions to install PyTorch on
No `neon` installation needed.
## Usage
The library defines `matmul` function similar to the one that
works with neon: ,
which instead of neon matrices takes `torch.cuda.FloatTensor` or `torch.cuda.HalfTensor`
as A, B and C.