Skip to content

Commit

Permalink
Fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
fmassa committed Sep 18, 2019
1 parent 8ad6930 commit ea893d3
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 2 deletions.
1 change: 0 additions & 1 deletion torchvision/csrc/ROIPool.h
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,6 @@ at::Tensor ROIPool_backward(
width);
}


using namespace at;
using torch::Tensor;
using torch::autograd::AutogradContext;
Expand Down
1 change: 0 additions & 1 deletion torchvision/csrc/vision.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,6 @@ PyMODINIT_FUNC PyInit__custom_ops(void) {
#endif
#endif


int64_t _cuda_version() {
#ifdef WITH_CUDA
return CUDA_VERSION;
Expand Down
3 changes: 3 additions & 0 deletions torchvision/extension.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
_HAS_OPS = False


def _register_extensions():
import os
import imp
Expand All @@ -10,6 +11,7 @@ def _register_extensions():
_, path, _ = imp.find_module("_C", [lib_dir])
torch.ops.load_library(path)


try:
_register_extensions()
_HAS_OPS = True
Expand Down Expand Up @@ -44,4 +46,5 @@ def _check_cuda_version():
.format(t_major, t_minor, tv_major, tv_minor))
return _version


_check_cuda_version()

0 comments on commit ea893d3

Please sign in to comment.