Python3 compatibility. (#39)

NVIDIA · Mar 21, 2018 · f33e07f · f33e07f
1 parent 208d4f6
commit f33e07f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/smooth_filter.py b/smooth_filter.py
@@ -329,8 +329,8 @@
 
 
 def smooth_local_affine(output_cpu, input_cpu, epsilon, patch, h, w, f_r, f_e):
-  program = Program(src, 'best_local_affine_kernel.cu')
-  ptx = program.compile(['-I/usr/local/cuda-9.1/include'])
+  program = Program(src.encode('utf-8'), 'best_local_affine_kernel.cu'.encode('utf-8'))
+  ptx = program.compile(['-I/usr/local/cuda-9.1/include'.encode('utf-8')])
   m = function.Module()
   m.load(bytes(ptx.encode()))