From 92f2d5272e30fa4d01880f001f7e55c5d8e6aab8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?No=C3=A9mien=20Kocher?= Date: Thu, 13 Dec 2018 20:36:41 +0100 Subject: [PATCH] Fixes wrong parameter call with the latest version of pynvrtc Leaving this fix here in case someone tries to run it with python3 and recent librairies. The error I encountered was ` 'Program' object has no attribute '_program'`, which was caused by the latest version of `pynvrtc`. More details: https://github.com/jonas-koehler/s2cnn/issues/21#issuecomment-409488734 --- 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: