From 2bc55f29efe9038384764b4bb44a6f458ac15294 Mon Sep 17 00:00:00 2001 From: vlado Date: Wed, 27 Feb 2019 10:02:42 -0700 Subject: [PATCH] Formatting fix --- python/mxnet/context.py | 2 +- src/storage/storage.cc | 9 +++------ 2 files changed, 4 insertions(+), 7 deletions(-) diff --git a/python/mxnet/context.py b/python/mxnet/context.py index ce48b9e6305f..1ca105146d01 100644 --- a/python/mxnet/context.py +++ b/python/mxnet/context.py @@ -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) diff --git a/src/storage/storage.cc b/src/storage/storage.cc index 809701ca3771..6eb8d1695672 100644 --- a/src/storage/storage.cc +++ b/src/storage/storage.cc @@ -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(); - } - else{ + } else { LOG(FATAL) << "Unknown memory pool strategy specified: " << strategy << "."; } #else