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

Commit

Permalink
Formatting fix
Browse files Browse the repository at this point in the history
  • Loading branch information
vlado committed Feb 27, 2019
1 parent 200d949 commit 2bc55f2
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 7 deletions.
2 changes: 1 addition & 1 deletion python/mxnet/context.py
Original file line number Diff line number Diff line change
Expand Up @@ -144,7 +144,7 @@ def default_ctx(cls, val):
DeprecationWarning)
cls._default_ctx.value = val
#pylint: enable=no-self-argument

def release_all(self):
dev_type = ctypes.c_int(self.device_typeid)
dev_id = ctypes.c_int(self.device_id)
Expand Down
9 changes: 3 additions & 6 deletions src/storage/storage.cc
Original file line number Diff line number Diff line change
Expand Up @@ -108,14 +108,11 @@ void StorageImpl::Alloc(Storage::Handle* handle) {
if (strategy == "Round") {
ptr = new storage::GPUPooledRoundedStorageManager();
LOG(INFO) << "Using GPUPooledRoundedStorageManager.";
}
else if (strategy == "Naive") {
} else if (strategy == "Naive") {
ptr = new storage::GPUPooledStorageManager();
}
else if(strategy == "Unpooled"){
} else if (strategy == "Unpooled") {
ptr = new storage::NaiveStorageManager<storage::GPUDeviceStorage>();
}
else{
} else {
LOG(FATAL) << "Unknown memory pool strategy specified: " << strategy << ".";
}
#else
Expand Down

0 comments on commit 2bc55f2

Please sign in to comment.