Add LAPACK functions

hongyunnchen · Oct 9, 2016 · 96f61bf · 96f61bf
1 parent f2cf673
commit 96f61bf
Show file tree

Hide file tree

Showing 10 changed files with 515 additions and 117 deletions.
diff --git a/setup.py b/setup.py
@@ -79,9 +79,10 @@ def run(self):
         from tools.cwrap.plugins.THPLongArgsPlugin import THPLongArgsPlugin
         from tools.cwrap.plugins.ArgcountSortPlugin import ArgcountSortPlugin
         from tools.cwrap.plugins.AutoGPU import AutoGPU
+        from tools.cwrap.plugins.BoolOption import BoolOption
         cwrap('torch/csrc/generic/TensorMethods.cwrap', plugins=[
-            AutoGPU(condition='IS_CUDA'), THPLongArgsPlugin(), THPPlugin(),
-            ArgcountSortPlugin(),
+            AutoGPU(condition='IS_CUDA'), THPLongArgsPlugin(), BoolOption(),
+            THPPlugin(), ArgcountSortPlugin(),
         ])
         # It's an old-style class in Python 2.7...
         setuptools.command.build_ext.build_ext.run(self)