CUda Matrix Multiply library
Project description
cumm
CUda Matrix Multiply library.
Install
Prebuilt
We offer python 3.6-3.10 and cuda 10.2/11.1/11.4 prebuilt binaries for linux (manylinux).
We offer python 3.7-3.10 and cuda 10.2/11.1/11.4 prebuilt binaries for windows 10/11.
We will offer prebuilts for CUDA versions supported by latest pytorch release. For example, pytorch 1.9 support cuda 10.2 and 11.1, so we support them too.
pip install cumm-cu102
for CUDA 10.2
pip install cumm-cu111
for CUDA 11.1
pip install cumm-cu114
for CUDA 11.4
Build from source
Linux
- install build-essential, install CUDA
- run
export CUMM_DISABLE_JIT="1"
- run
python setup.py install
/pip install -e .
/python setup.py bdist_wheel
+pip install dists/xxx.whl
Windows 10/11
- install visual studio 2019 or newer. make sure C++ development package is installed. install CUDA
- set powershell script execution policy
- start a new powershell, run
tools/msvc_setup.ps1
- run
$Env:CUMM_DISABLE_JIT = "1"
- run
python setup.py install
/pip install -e .
/python setup.py bdist_wheel
+pip install dists/xxx.whl
Note
The work is done when the author is an employee at Tusimple.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distributions
Close
Hashes for cumm_cu102-0.1.10-cp310-cp310-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 93224b32319c23b3d09b17f8cc30d283732aebef1429dff8175e8a57419ff2d7 |
|
MD5 | 6d9c3451767ed5ac46ad4eba0ec66512 |
|
BLAKE2b-256 | 17797c99d0c46dc30a600d63886ad70327957b769fa67a1993578f476a06d775 |
Close
Hashes for cumm_cu102-0.1.10-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0b99e0fd6d314ee287d245dba8f49931e27c36451706470d8833575d0a9517e4 |
|
MD5 | 1c1948ea55bfd5d4cf2ff87f9959d1f7 |
|
BLAKE2b-256 | 7b70e484336275057b49d0f04f15d7bd5d2e7f6c56341281d55f8374d4c1676c |
Close
Hashes for cumm_cu102-0.1.10-cp39-cp39-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 25aad927b2fa38242745e6d11519a530626a9436406f22f7ce40fa6a06fe04a6 |
|
MD5 | 3a5b711202571773d463fe4e5d631909 |
|
BLAKE2b-256 | b02bf2956ab292a9210ae6bdec43426d2be3096732bd3e1b2925e6d3c0834049 |
Close
Hashes for cumm_cu102-0.1.10-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 65e9e5186f5c10667b2ff3b0e5f41a5cd0a48715424c5af625b8317862dde83c |
|
MD5 | 8b0c2084b209414a42d9764f6796fe47 |
|
BLAKE2b-256 | 2c07951f83aa4e3d011f08f2ccd1936a3823d3ef781421ee004d78ad3416038b |
Close
Hashes for cumm_cu102-0.1.10-cp38-cp38-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ea42d81f128d5298b84c1a8005019465065915ca15b91b0f8e9073a29a6bccb7 |
|
MD5 | 169d81cb65768d927c8ad2f59e059b48 |
|
BLAKE2b-256 | a777c133ced9e261856d190995d97ed2553af90790b35c920ed96a70cf3a8fa3 |
Close
Hashes for cumm_cu102-0.1.10-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4d1cd8ecbc404ce54927e52e011fdc23313d77a16628206fdb41a4aa1e4eb160 |
|
MD5 | 5e8e0e00fae1a0e144ec666cdde9e1e2 |
|
BLAKE2b-256 | 39ab8132b857098e658ed352f7854f2c0a30ba88ef382ffb7ce6933bcb0ea184 |
Close
Hashes for cumm_cu102-0.1.10-cp37-cp37m-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d31e846cf0abced0a581664353f807b81d851258651a40ffa5e02f5ae24de287 |
|
MD5 | 6f776977b1c69a472e9bffbb962fe0e7 |
|
BLAKE2b-256 | 0990e3bfa585fb96090fa352a6cd44d3eb0032838fed78921b5b98f4ed3e6607 |
Close
Hashes for cumm_cu102-0.1.10-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 93948278f89b90fb76df8373995ee4116feb02fa8def348e2b055262c8278eaf |
|
MD5 | 039b05ebb991526d73a467edead9d52a |
|
BLAKE2b-256 | c5a5ffac1c6e6af207570a456773d63e0e1d73b6032dde8dd2a90d1369ae1d4a |
Close
Hashes for cumm_cu102-0.1.10-cp36-cp36m-win_amd64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5671d7d0fce52ee6a9768a8c3973ff335cfe3c8cfce15806287ae4abd9a058d2 |
|
MD5 | 2234593eedc9b9457dea41c70d633ebe |
|
BLAKE2b-256 | eb87a23a6613b2e118655eddbb1c49cce75a51318982cbb8e45e27313c3c0068 |
Close
Hashes for cumm_cu102-0.1.10-cp36-cp36m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 62b8eb2ac299b5bc5a458274326da52e58d3044defb61ccb85c35cda25369f88 |
|
MD5 | 4eb601a3400cca5d1d60b37f59dcc132 |
|
BLAKE2b-256 | f5eda388b20d3f93bbf7ef8b121b40a4b4c0199d9f54581a27246c8cacf10e7b |