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

Allow support for negative dimension argument for all functions

Browse Source
This commit is contained in:
albanD
2017-03-26 14:25:53 +01:00
committed by soumith
parent e7f5220dfa
commit f0c7124420
13 changed files with 394 additions and 179 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@ -154,10 +154,11 @@ class build_ext(setuptools.command.build_ext.build_ext):
from tools.cwrap.plugins.KwargsPlugin import KwargsPlugin
from tools.cwrap.plugins.NullableArguments import NullableArguments
from tools.cwrap.plugins.CuDNNPlugin import CuDNNPlugin
from tools.cwrap.plugins.WrapDim import WrapDim
thp_plugin = THPPlugin()
cwrap('torch/csrc/generic/TensorMethods.cwrap', plugins=[
BoolOption(), thp_plugin, AutoGPU(condition='IS_CUDA'),
ArgcountSortPlugin(), KwargsPlugin()
ArgcountSortPlugin(), KwargsPlugin(), WrapDim()
])
cwrap('torch/csrc/cudnn/cuDNN.cwrap', plugins=[
CuDNNPlugin(), NullableArguments()