lua 5.2 compat

1 files changed, 7 insertions, 0 deletions

diff --git a/init.lua b/init.lua
index 5108a16..92640b1 100644
--- a/init.lua
+++ b/init.lua
@@ -1,5 +1,12 @@
 
 -- We are using paths.require to appease mkl
+
+-- Make this work with LuaJIT in Lua 5.2 compatibility mode, which
+-- renames string.gfind (already deprecated in 5.1)
+if not string.gfind then
+    string.gfind = string.gmatch
+end
+
 require "paths"
 paths.require "libtorch"