pytorch 1.0.0

rusty1s · Jan 23, 2019 · 1d135a2 · 1d135a2
1 parent e21ee09
commit 1d135a2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.coveragerc b/.coveragerc
@@ -1,3 +1,5 @@
+[run]
+source=torch_scatter
 [report]
 exclude_lines =
     pragma: no cover

diff --git a/LICENSE b/LICENSE
@@ -1,4 +1,4 @@
-Copyright (c) 2018 Matthias Fey <matthias.fey@tu-dortmund.de>
+Copyright (c) 2019 Matthias Fey <matthias.fey@tu-dortmund.de>
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

diff --git a/cuda/scatter.cpp b/cuda/scatter.cpp
@@ -1,4 +1,4 @@
-#include <torch/torch.h>
+#include <torch/extension.h>
 
 #define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be CUDA tensor")