From f6f147a777d000745457a567f3702b9ad7870e9f Mon Sep 17 00:00:00 2001 From: AntoinePrv Date: Fri, 23 Nov 2018 10:17:40 -0500 Subject: [PATCH 1/2] Weaker test for cuda extension. --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 8a14277f..78db09c9 100644 --- a/setup.py +++ b/setup.py @@ -1,6 +1,6 @@ from setuptools import setup, find_packages import torch -from torch.utils.cpp_extension import BuildExtension, CUDAExtension +from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CUDA_HOME __version__ = '0.2.2' url = 'https://github.com/rusty1s/pytorch_sparse' @@ -11,7 +11,7 @@ ext_modules = [] cmdclass = {} -if torch.cuda.is_available(): +if CUDA_HOME is not None: ext_modules += [ CUDAExtension( 'spspmm_cuda', From 84d5c89e8c2b850ffbc2ad1d1282599556d6fbc0 Mon Sep 17 00:00:00 2001 From: AntoinePrv Date: Fri, 23 Nov 2018 10:18:44 -0500 Subject: [PATCH 2/2] Deprecation warning --- torch_sparse/spspmm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_sparse/spspmm.py b/torch_sparse/spspmm.py index 6f5d4d3a..9ed5afee 100644 --- a/torch_sparse/spspmm.py +++ b/torch_sparse/spspmm.py @@ -37,7 +37,7 @@ def forward(ctx, indexA, valueA, indexB, valueB, m, k, n): @staticmethod def backward(ctx, grad_indexC, grad_valueC): m, k, n = ctx.m, ctx.k, ctx.n - indexA, valueA, indexB, valueB, indexC = ctx.saved_variables + indexA, valueA, indexB, valueB, indexC = ctx.saved_tensors grad_valueA = grad_valueB = None