Skip to content

Commit a8df961

Browse files
committed
fix setup
1 parent dff1c0e commit a8df961

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

pytorch/edgeml_pytorch/graph/rnn.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -324,8 +324,6 @@ def __init__(self, input_size, hidden_size, zetaInit=1.0, nuInit=-4.0, name="Fas
324324
raise Exception('FastGRNNCUDACell is supported only on GPU devices.')
325325
self._input_size = input_size
326326
self._hidden_size = hidden_size
327-
self._gate_non_linearity = gate_non_linearity
328-
self._update_non_linearity = update_non_linearity
329327
self._zetaInit = zetaInit
330328
self._nuInit = nuInit
331329
self._name = name

pytorch/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from edgeml_pytorch.utils import findCUDA
55

66
if findCUDA() is not None:
7-
setup(
7+
setuptools.setup(
88
name='fastgrnn_cuda',
99
ext_modules=[
1010
CUDAExtension('fastgrnn_cuda', [

0 commit comments

Comments
 (0)