frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

use "Extension" instead of the unimported "setuptools.Extension" (#14475)

Browse Source 
Summary:
use "Extension" instead of the unimported "setuptools.Extension"
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14475

Differential Revision: D13356219

Pulled By: ezyang

fbshipit-source-id: 5a3e7eb73a32d6bf09676efd9eddded5586435cd
This commit is contained in:
HB_alon
2018-12-05 22:16:44 -08:00
committed by Facebook Github Bot
parent d393dd0744
commit 5e307bd1be
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -901,13 +901,13 @@ if USE_CUDA:
# These extensions are built by cmake and copied manually in build_extensions()
# inside the build_ext implementaiton
extensions.append(
setuptools.Extension(
Extension(
name=str('caffe2.python.caffe2_pybind11_state'),
sources=[]),
)
if USE_CUDA:
extensions.append(
setuptools.Extension(
Extension(
name=str('caffe2.python.caffe2_pybind11_state_gpu'),
sources=[]),
)