elephant-track / elephant-server

A server implementation of ELEPHANT

BSD 2-Clause "Simplified" License

7 stars 5 forks source link

ELEPHANT server v0.3.2 -> v0.3.3 #12

Closed ksugar closed 2 years ago

ksugar commented 2 years ago

ELEPHANT server v0.3.3

Updates

Update PyTorch (1.4.0 -> 1.10.0), CudaToolKit (10.1.243 -> 11.3) and cuDNN (7.6.3 -> 8.2.0)
- support Ampere architecture
Use the same parameters for allclose in Numpy and PyTorch
- Use the default parameters for PyTorch (rtol=1e-05, atol=1e-08)
- exceptionally, the rtol values in test_models.py are set to 1e-4
- assert is added to test_models.py