Fast Forward DSSTNE to NVLINK and faster sparse kernels

amazon-archives / amazon-dsstne

Deep Scalable Sparse Tensor Network Engine (DSSTNE) is an Amazon developed library for building Deep Learning (DL) machine learning (ML) models

Apache License 2.0

4.41k stars 729 forks source link

Open scottlegrand opened 4 years ago

scottlegrand commented 4 years ago

Issue #, if available: No issue

*Description of changes: Adds NVLINK support through DGX-1V, and enables more efficient sparse kernels for smaller embedding widths.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.