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 731 forks source link

Set minValue to the new min after bitonic sort to properly filter val… #212

Closed kiukA9 closed 5 years ago

kiukA9 commented 5 years ago

Description of changes: After bitonic sort is called (when the shared mem buffer gets full), we should set the minVal equal to the register that contains the smallest value (the largest numbered register since the sorting is in descending order). This has the effect of filtering out values that are smaller than the running k^th element, hence reducing the number of times we call bitonic sort.

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