1 files changed, 21 insertions, 11 deletions

diff --git a/utils/hwstub/tools/lua/load.lua b/utils/hwstub/tools/lua/load.lua
index f636360c4c..cafa82751e 100644
--- a/utils/hwstub/tools/lua/load.lua
+++ b/utils/hwstub/tools/lua/load.lua
@@ -1,14 +1,24 @@
 package.path = string.sub(string.gsub(debug.getinfo(1).source, "load.lua", "?.lua"),2) .. ";" .. package.path
-
-if hwstub.dev.target.id == hwstub.dev.target.STMP then
-    require "stmp"
-elseif hwstub.dev.target.id == hwstub.dev.target.PP then
-    require "pp"
-elseif hwstub.dev.target.id == hwstub.dev.target.RK27 then
-    require "rk27xx"
-elseif hwstub.dev.target.id == hwstub.dev.target.ATJ then
-    require "atj"
-end
-
+require "stmp"
+require "pp"
+require "rk27xx"
+require "atj"
+require "jz"
 require "hwlib"
 require "dumper"
+
+LOAD = {}
+
+function LOAD.init()
+    if hwstub.dev.target.id == hwstub.dev.target.STMP then
+        STMP.init()
+    elseif hwstub.dev.target.id == hwstub.dev.target.PP then
+        PP.init()
+    elseif hwstub.dev.target.id == hwstub.dev.target.RK27 then
+        RK27XX.init()
+    elseif hwstub.dev.target.id == hwstub.dev.target.ATJ then
+        ATJ.init()
+    elseif hwstub.dev.target.id == hwstub.dev.target.JZ then
+        JZ.init()
+    end
+end
\ No newline at end of file