From 7d272c4277fc47b4c5a85085296a2f167e09d6b1 Mon Sep 17 00:00:00 2001 From: Aleksandr Seliverstov Date: Wed, 26 Sep 2018 15:33:47 +0300 Subject: [PATCH] fix: remove duplicate `encode` Issue described in https://github.com/salesforce/pytorch-qrnn/issues/17 --- torchqrnn/forget_mult.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchqrnn/forget_mult.py b/torchqrnn/forget_mult.py index 6967850..9c07119 100644 --- a/torchqrnn/forget_mult.py +++ b/torchqrnn/forget_mult.py @@ -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: