Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Refactor Storage -> FeatureSet
Browse files Browse the repository at this point in the history
  • Loading branch information
larroy committed Jan 9, 2019
1 parent 93529d3 commit 7aff860
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/mxfeatures.cc
Original file line number Diff line number Diff line change
Expand Up @@ -32,9 +32,9 @@ namespace mxnet {
namespace features {


class Storage {
class FeatureSet {
public:
Storage():
FeatureSet():
feature_bits() {
// GPU
feature_bits.set(CUDA, MXNET_USE_CUDA);
Expand Down Expand Up @@ -105,10 +105,10 @@ class Storage {
std::bitset<MAX_FEATURES> feature_bits;
};

static Storage storage;
static FeatureSet featureSet;

bool is_enabled(const unsigned feat) {
return storage.is_enabled(feat);
return featureSet.is_enabled(feat);
}

} // namespace features
Expand Down

0 comments on commit 7aff860

Please sign in to comment.