pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Jorghi12 81b92f7515 Get ROCm building again on master (#8343 )

Billing of changes:

- New Jenkins script for building on rocm. For now it is a bit hacked together, but we can improve it once CI is running
- New ROCM docker image for nightly HIP, and also some legacy packages that we need temporarily
- New enabled config py2-clang3.8-rocmnightly-ubuntu16.04-build based off of the existing Caffe2 image (not built yet)
- A big pile of cmake fixes, mostly to turn bits on/off when ROCM build is involved
- Switch from hiprng to hcrng
- Apply some patches directly in code, eliminating the patches
- Use __hdiv instead of hdiv, it's more portable
- THCNumerics<T>::gt doesn't work in HIP, so simulate it with sub
- Add a few more overloads HIP needs
- Turn off use of hcc to link (we plan to turn this back on to get tests running)
- Search for hiprand, hiprng, hipblas, hipsparse
- Better Python 2 portability

2018-06-12 23:05:21 -04:00

__init__.py

Improve cuDNN detection at build time

2016-12-01 23:14:41 +01:00

cuda.py

[build] remove the use of NO_CUDA (#8300 )

2018-06-12 12:14:36 -04:00

cudnn.py

[pytorch] add static linkage support for CuDNN and NCCL (#6410 )

2018-04-08 22:54:18 -04:00

dist_check.py

Create initial Python bindings for c10d (#8119 )