bugfix for legacy shuffle API

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

Closed jeng1220 closed 4 years ago

jeng1220 commented 4 years ago

Issue #, if available: #227

Description of changes:

Replace __shfl with SHFL to keep CUDA shuffle API compatible
Update README.md to guide user to download new version CUB library because old CUB library uses legacy CUDA shuffle API

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