Skip to content

Commit

Permalink
fix: remove duplicate encode
Browse files Browse the repository at this point in the history
Issue described in
salesforce#17
  • Loading branch information
seliverstov authored Sep 26, 2018
1 parent daadb0f commit 7d272c4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchqrnn/forget_mult.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ def __init__(self):

def compile(self):
if self.ptx is None:
program = Program(kernel.encode(), 'recurrent_forget_mult.cu'.encode())
program = Program(kernel, 'recurrent_forget_mult.cu')
GPUForgetMult.ptx = program.compile()

if torch.cuda.current_device() not in GPUForgetMult.configured_gpus:
Expand Down

0 comments on commit 7d272c4

Please sign in to comment.