From 3e477fcf18d05c40ee268c434bb5e6d403b56998 Mon Sep 17 00:00:00 2001 From: Russi Chatterjee Date: Fri, 5 Apr 2019 15:22:48 +0530 Subject: [PATCH] fix #36 --- dnc/util.py | 6 +++--- setup.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/dnc/util.py b/dnc/util.py index 87ae5d6..e16e0be 100644 --- a/dnc/util.py +++ b/dnc/util.py @@ -29,7 +29,7 @@ def cuda(x, grad=False, gpu_id=-1): t.requires_grad=grad return t else: - t = T.FloatTensor(x.pin_memory()).cuda(gpu_id, async=True) + t = T.FloatTensor(x.pin_memory()).cuda(gpu_id) t.requires_grad=grad return t @@ -40,7 +40,7 @@ def cudavec(x, grad=False, gpu_id=-1): t.requires_grad = grad return t else: - t = T.Tensor(T.from_numpy(x).pin_memory()).cuda(gpu_id, async=True) + t = T.Tensor(T.from_numpy(x).pin_memory()).cuda(gpu_id) t.requires_grad = grad return t @@ -51,7 +51,7 @@ def cudalong(x, grad=False, gpu_id=-1): t.requires_grad = grad return t else: - t = T.LongTensor(T.from_numpy(x.astype(np.long)).pin_memory()).cuda(gpu_id, async=True) + t = T.LongTensor(T.from_numpy(x.astype(np.long)).pin_memory()).cuda(gpu_id) t.requires_grad = grad return t diff --git a/setup.py b/setup.py index c076a51..9e47555 100644 --- a/setup.py +++ b/setup.py @@ -22,7 +22,7 @@ with open(path.join(here, 'README.rst'), encoding='utf-8') as f: setup( name='dnc', - version='1.0.0', + version='1.0.1', description='Differentiable Neural Computer, for Pytorch', long_description=long_description,