format files and adjust architecture

.gitignore

@@ -96,3 +96,6 @@ metal/paddle-mobile/paddle-mobile/CPU/libpaddle-mobile.a
 
 metal/paddle-mobile-demo/paddle-mobile-demo/images
 metal/paddle-mobile-demo/paddle-mobile-demo/models
+metal/paddle-mobile-demo/paddle-mobile-demo/Resources
+metal/paddle-mobile-demo/paddle-mobile-demo/Resources/images
+metal/paddle-mobile-demo/paddle-mobile-demo/Resources/models

metal/paddle-mobile-demo/paddle-mobile-demo.xcodeproj/project.pbxproj

@@ -10,13 +10,26 @@
 		30D0ED21F392CFA3885B1002 /* Pods_paddle_mobile_demo.framework in Frameworks */ = {isa = PBXBuildFile; fileRef = 18896810981724F8A0FED62A /* Pods_paddle_mobile_demo.framework */; };
 		C2CBB49021B778EA0020DC6C /* libc++.tbd in Frameworks */ = {isa = PBXBuildFile; fileRef = FC4FD97B2140EE250073E130 /* libc++.tbd */; };
 		C2E67E5E21524E460013F575 /* LoadPointerViewController.m in Sources */ = {isa = PBXBuildFile; fileRef = C2E67E5D21524E460013F575 /* LoadPointerViewController.m */; };
-		FC013928210204A3008100E3 /* PreProcessKernel.metal in Sources */ = {isa = PBXBuildFile; fileRef = FC013927210204A3008100E3 /* PreProcessKernel.metal */; };
 		FC039B8220E11C550081E9F8 /* AppDelegate.swift in Sources */ = {isa = PBXBuildFile; fileRef = FC039B8120E11C550081E9F8 /* AppDelegate.swift */; };
 		FC039B8420E11C550081E9F8 /* ViewController.swift in Sources */ = {isa = PBXBuildFile; fileRef = FC039B8320E11C550081E9F8 /* ViewController.swift */; };
 		FC039B8720E11C550081E9F8 /* Main.storyboard in Resources */ = {isa = PBXBuildFile; fileRef = FC039B8520E11C550081E9F8 /* Main.storyboard */; };
 		FC039B8920E11C560081E9F8 /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = FC039B8820E11C560081E9F8 /* Assets.xcassets */; };
 		FC039B8C20E11C560081E9F8 /* LaunchScreen.storyboard in Resources */ = {isa = PBXBuildFile; fileRef = FC039B8A20E11C560081E9F8 /* LaunchScreen.storyboard */; };
 		FC203FB221CBFDBA00B37166 /* test.jpg in Resources */ = {isa = PBXBuildFile; fileRef = FC203FA921CBFDBA00B37166 /* test.jpg */; };
+		FC2BFCBC21DF0A8600C262B2 /* 00001.jpg in Resources */ = {isa = PBXBuildFile; fileRef = FC2BFCBB21DF0A8600C262B2 /* 00001.jpg */; };
+		FC2BFCBE21DF15D900C262B2 /* 123.jpg in Resources */ = {isa = PBXBuildFile; fileRef = FC2BFCBD21DF15D900C262B2 /* 123.jpg */; };
+		FC2BFCC021DF279900C262B2 /* classify-img-output.png in Resources */ = {isa = PBXBuildFile; fileRef = FC2BFCBF21DF279900C262B2 /* classify-img-output.png */; };
+		FC2BFD3021DF3FEA00C262B2 /* MobilenetSSD_AR.swift in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD2A21DF3FE900C262B2 /* MobilenetSSD_AR.swift */; };
+		FC2BFD3121DF3FEA00C262B2 /* Genet.swift in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD2B21DF3FE900C262B2 /* Genet.swift */; };
+		FC2BFD3221DF3FEA00C262B2 /* MobileNetSSD.swift in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD2C21DF3FE900C262B2 /* MobileNetSSD.swift */; };
+		FC2BFD3321DF3FEA00C262B2 /* YoloNet.swift in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD2D21DF3FE900C262B2 /* YoloNet.swift */; };
+		FC2BFD3421DF3FEA00C262B2 /* MobileNetCombined.swift in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD2E21DF3FEA00C262B2 /* MobileNetCombined.swift */; };
+		FC2BFD3521DF3FEA00C262B2 /* MobileNet.swift in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD2F21DF3FEA00C262B2 /* MobileNet.swift */; };
+		FC2BFD3821DF46DE00C262B2 /* OCDemoViewController.m in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD3721DF46DE00C262B2 /* OCDemoViewController.m */; };
+		FC2BFD3C21DF480400C262B2 /* CPUCompute.mm in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD3B21DF480400C262B2 /* CPUCompute.mm */; };
+		FC2BFD3E21DF5CE800C262B2 /* PreProcessKernel.metal in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD3D21DF5CE800C262B2 /* PreProcessKernel.metal */; };
+		FC2BFD4321DF5E1E00C262B2 /* PaddleMobileGPU.m in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD4021DF5E1E00C262B2 /* PaddleMobileGPU.m */; };
+		FC2BFD4421DF5E1E00C262B2 /* SuperResolutionNet.swift in Sources */ = {isa = PBXBuildFile; fileRef = FC2BFD4221DF5E1E00C262B2 /* SuperResolutionNet.swift */; };
 		FC5E03B221DCE8D90016C137 /* mingren_input_data in Resources */ = {isa = PBXBuildFile; fileRef = FC5E03B121DCE8D90016C137 /* mingren_input_data */; };
 		FC704C1921D2375300F98BAB /* super_params in Resources */ = {isa = PBXBuildFile; fileRef = FC704C1721D2375300F98BAB /* super_params */; };
 		FC704C1A21D2375300F98BAB /* super_model in Resources */ = {isa = PBXBuildFile; fileRef = FC704C1821D2375300F98BAB /* super_model */; };
@@ -31,7 +44,6 @@
 		FC9797C321D608E000F2FD90 /* mobilenet_params in Resources */ = {isa = PBXBuildFile; fileRef = FC9797C121D608DF00F2FD90 /* mobilenet_params */; };
 		FC9797C721D609FB00F2FD90 /* synset.txt in Resources */ = {isa = PBXBuildFile; fileRef = FC9797C621D609FB00F2FD90 /* synset.txt */; };
 		FC9797CF21D6506F00F2FD90 /* mingren.jpg in Resources */ = {isa = PBXBuildFile; fileRef = FC9797CE21D6506F00F2FD90 /* mingren.jpg */; };
-		FC9797D121D6616600F2FD90 /* BufferToTexture.metal in Sources */ = {isa = PBXBuildFile; fileRef = FC9797D021D6616600F2FD90 /* BufferToTexture.metal */; };
 		FCBCCC552122EF5500D94F7E /* MetalHelper.swift in Sources */ = {isa = PBXBuildFile; fileRef = FCBCCC542122EF5400D94F7E /* MetalHelper.swift */; };
 		FCCED60521D7646E00BE8D5F /* test_image_super in Resources */ = {isa = PBXBuildFile; fileRef = FCCED60421D7646E00BE8D5F /* test_image_super */; };
 		FCEBEC2C20E1391F00C0B14D /* paddle_mobile.framework in Frameworks */ = {isa = PBXBuildFile; fileRef = FCEBEC2B20E1391F00C0B14D /* paddle_mobile.framework */; };
@@ -59,7 +71,6 @@
 		878829884E1A14D7044721D5 /* Pods-paddle-mobile-demo.debug.xcconfig */ = {isa = PBXFileReference; includeInIndex = 1; lastKnownFileType = text.xcconfig; name = "Pods-paddle-mobile-demo.debug.xcconfig"; path = "../Pods/Target Support Files/Pods-paddle-mobile-demo/Pods-paddle-mobile-demo.debug.xcconfig"; sourceTree = "<group>"; };
 		C2E67E5C21524E460013F575 /* LoadPointerViewController.h */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.h; path = LoadPointerViewController.h; sourceTree = "<group>"; };
 		C2E67E5D21524E460013F575 /* LoadPointerViewController.m */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.objc; path = LoadPointerViewController.m; sourceTree = "<group>"; };
-		FC013927210204A3008100E3 /* PreProcessKernel.metal */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.metal; path = PreProcessKernel.metal; sourceTree = "<group>"; };
 		FC039B7E20E11C550081E9F8 /* paddle-mobile-demo.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = "paddle-mobile-demo.app"; sourceTree = BUILT_PRODUCTS_DIR; };
 		FC039B8120E11C550081E9F8 /* AppDelegate.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = AppDelegate.swift; sourceTree = "<group>"; };
 		FC039B8320E11C550081E9F8 /* ViewController.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ViewController.swift; sourceTree = "<group>"; };
@@ -69,6 +80,23 @@
 		FC039B8D20E11C560081E9F8 /* Info.plist */ = {isa = PBXFileReference; lastKnownFileType = text.plist.xml; path = Info.plist; sourceTree = "<group>"; };
 		FC203FA921CBFDBA00B37166 /* test.jpg */ = {isa = PBXFileReference; lastKnownFileType = image.jpeg; path = test.jpg; sourceTree = "<group>"; };
 		FC27991121343A39000B6BAD /* paddle-mobile-demo-Bridging-Header.h */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.h; path = "paddle-mobile-demo-Bridging-Header.h"; sourceTree = "<group>"; };
+		FC2BFCBB21DF0A8600C262B2 /* 00001.jpg */ = {isa = PBXFileReference; lastKnownFileType = image.jpeg; path = 00001.jpg; sourceTree = "<group>"; };
+		FC2BFCBD21DF15D900C262B2 /* 123.jpg */ = {isa = PBXFileReference; lastKnownFileType = image.jpeg; path = 123.jpg; sourceTree = "<group>"; };
+		FC2BFCBF21DF279900C262B2 /* classify-img-output.png */ = {isa = PBXFileReference; lastKnownFileType = image.png; path = "classify-img-output.png"; sourceTree = "<group>"; };
+		FC2BFD2A21DF3FE900C262B2 /* MobilenetSSD_AR.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = MobilenetSSD_AR.swift; sourceTree = "<group>"; };
+		FC2BFD2B21DF3FE900C262B2 /* Genet.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = Genet.swift; sourceTree = "<group>"; };
+		FC2BFD2C21DF3FE900C262B2 /* MobileNetSSD.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = MobileNetSSD.swift; sourceTree = "<group>"; };
+		FC2BFD2D21DF3FE900C262B2 /* YoloNet.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = YoloNet.swift; sourceTree = "<group>"; };
+		FC2BFD2E21DF3FEA00C262B2 /* MobileNetCombined.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = MobileNetCombined.swift; sourceTree = "<group>"; };
+		FC2BFD2F21DF3FEA00C262B2 /* MobileNet.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = MobileNet.swift; sourceTree = "<group>"; };
+		FC2BFD3621DF46DE00C262B2 /* OCDemoViewController.h */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.h; path = OCDemoViewController.h; sourceTree = "<group>"; };
+		FC2BFD3721DF46DE00C262B2 /* OCDemoViewController.m */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.objc; path = OCDemoViewController.m; sourceTree = "<group>"; };
+		FC2BFD3A21DF480300C262B2 /* CPUCompute.h */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.c.h; path = CPUCompute.h; sourceTree = "<group>"; };
+		FC2BFD3B21DF480400C262B2 /* CPUCompute.mm */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.cpp.objcpp; path = CPUCompute.mm; sourceTree = "<group>"; };
+		FC2BFD3D21DF5CE800C262B2 /* PreProcessKernel.metal */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.metal; path = PreProcessKernel.metal; sourceTree = "<group>"; };
+		FC2BFD4021DF5E1E00C262B2 /* PaddleMobileGPU.m */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.c.objc; path = PaddleMobileGPU.m; sourceTree = "<group>"; };
+		FC2BFD4121DF5E1E00C262B2 /* PaddleMobileGPU.h */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.c.h; path = PaddleMobileGPU.h; sourceTree = "<group>"; };
+		FC2BFD4221DF5E1E00C262B2 /* SuperResolutionNet.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = SuperResolutionNet.swift; sourceTree = "<group>"; };
 		FC4FD97B2140EE250073E130 /* libc++.tbd */ = {isa = PBXFileReference; lastKnownFileType = "sourcecode.text-based-dylib-definition"; name = "libc++.tbd"; path = "usr/lib/libc++.tbd"; sourceTree = SDKROOT; };
 		FC5E03B121DCE8D90016C137 /* mingren_input_data */ = {isa = PBXFileReference; lastKnownFileType = file; path = mingren_input_data; sourceTree = "<group>"; };
 		FC704C1721D2375300F98BAB /* super_params */ = {isa = PBXFileReference; lastKnownFileType = file; path = super_params; sourceTree = "<group>"; };
@@ -84,7 +112,6 @@
 		FC9797C121D608DF00F2FD90 /* mobilenet_params */ = {isa = PBXFileReference; lastKnownFileType = file; path = mobilenet_params; sourceTree = "<group>"; };
 		FC9797C621D609FB00F2FD90 /* synset.txt */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = text; path = synset.txt; sourceTree = "<group>"; };
 		FC9797CE21D6506F00F2FD90 /* mingren.jpg */ = {isa = PBXFileReference; lastKnownFileType = image.jpeg; path = mingren.jpg; sourceTree = "<group>"; };
-		FC9797D021D6616600F2FD90 /* BufferToTexture.metal */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.metal; path = BufferToTexture.metal; sourceTree = "<group>"; };
 		FCBCCC542122EF5400D94F7E /* MetalHelper.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = MetalHelper.swift; sourceTree = "<group>"; };
 		FCCED60421D7646E00BE8D5F /* test_image_super */ = {isa = PBXFileReference; lastKnownFileType = file; path = test_image_super; sourceTree = "<group>"; };
 		FCEBEC2B20E1391F00C0B14D /* paddle_mobile.framework */ = {isa = PBXFileReference; explicitFileType = wrapper.framework; path = paddle_mobile.framework; sourceTree = BUILT_PRODUCTS_DIR; };
@@ -145,8 +172,11 @@
 		FC039B8020E11C550081E9F8 /* paddle-mobile-demo */ = {
 			isa = PBXGroup;
 			children = (
-				FC203FA821CBFDBA00B37166 /* images */,
-				FC203FAA21CBFDBA00B37166 /* models */,
+				FC2BFD4F21DF892500C262B2 /* Resources */,
+				FCBCCC542122EF5400D94F7E /* MetalHelper.swift */,
+				FC2BFD3F21DF5DDF00C262B2 /* OCInterface */,
+				FC27991121343A39000B6BAD /* paddle-mobile-demo-Bridging-Header.h */,
+				FC2BFD3921DF46F000C262B2 /* OCDemo */,
 				FC803BCA214D27920094B8E5 /* VideoCapture */,
 				FC8CFED2213519540094D569 /* Net */,
 				FC039B8120E11C550081E9F8 /* AppDelegate.swift */,
@@ -155,10 +185,7 @@
 				FC039B8820E11C560081E9F8 /* Assets.xcassets */,
 				FC039B8A20E11C560081E9F8 /* LaunchScreen.storyboard */,
 				FC039B8D20E11C560081E9F8 /* Info.plist */,
-				FC27991121343A39000B6BAD /* paddle-mobile-demo-Bridging-Header.h */,
 				FCF437E7214B6DDB00943429 /* MultiPredictViewController.swift */,
-				C2E67E5C21524E460013F575 /* LoadPointerViewController.h */,
-				C2E67E5D21524E460013F575 /* LoadPointerViewController.m */,
 			);
 			path = "paddle-mobile-demo";
 			sourceTree = "<group>";
@@ -166,14 +193,16 @@
 		FC203FA821CBFDBA00B37166 /* images */ = {
 			isa = PBXGroup;
 			children = (
+				FC2BFCBF21DF279900C262B2 /* classify-img-output.png */,
+				FC2BFCBD21DF15D900C262B2 /* 123.jpg */,
+				FC2BFCBB21DF0A8600C262B2 /* 00001.jpg */,
 				FC5E03B121DCE8D90016C137 /* mingren_input_data */,
 				FCCED60421D7646E00BE8D5F /* test_image_super */,
 				FC9797CE21D6506F00F2FD90 /* mingren.jpg */,
 				FC9797BD21D6045B00F2FD90 /* banana.jpeg */,
 				FC203FA921CBFDBA00B37166 /* test.jpg */,
 			);
-			name = images;
-			path = ../../images;
+			path = images;
 			sourceTree = "<group>";
 		};
 		FC203FAA21CBFDBA00B37166 /* models */ = {
@@ -183,8 +212,37 @@
 				FC704C1B21D237FC00F98BAB /* vision_model */,
 				FC704C1621D2375300F98BAB /* superresoltion */,
 			);
-			name = models;
-			path = ../../models;
+			path = models;
+			sourceTree = "<group>";
+		};
+		FC2BFD3921DF46F000C262B2 /* OCDemo */ = {
+			isa = PBXGroup;
+			children = (
+				C2E67E5C21524E460013F575 /* LoadPointerViewController.h */,
+				C2E67E5D21524E460013F575 /* LoadPointerViewController.m */,
+				FC2BFD3621DF46DE00C262B2 /* OCDemoViewController.h */,
+				FC2BFD3721DF46DE00C262B2 /* OCDemoViewController.m */,
+			);
+			path = OCDemo;
+			sourceTree = "<group>";
+		};
+		FC2BFD3F21DF5DDF00C262B2 /* OCInterface */ = {
+			isa = PBXGroup;
+			children = (
+				FC2BFD4121DF5E1E00C262B2 /* PaddleMobileGPU.h */,
+				FC2BFD4021DF5E1E00C262B2 /* PaddleMobileGPU.m */,
+				FC2BFD4221DF5E1E00C262B2 /* SuperResolutionNet.swift */,
+			);
+			path = OCInterface;
+			sourceTree = "<group>";
+		};
+		FC2BFD4F21DF892500C262B2 /* Resources */ = {
+			isa = PBXGroup;
+			children = (
+				FC203FA821CBFDBA00B37166 /* images */,
+				FC203FAA21CBFDBA00B37166 /* models */,
+			);
+			path = Resources;
 			sourceTree = "<group>";
 		};
 		FC704C1621D2375300F98BAB /* superresoltion */ = {
@@ -235,9 +293,15 @@
 		FC8CFED2213519540094D569 /* Net */ = {
 			isa = PBXGroup;
 			children = (
-				FC013927210204A3008100E3 /* PreProcessKernel.metal */,
-				FCBCCC542122EF5400D94F7E /* MetalHelper.swift */,
-				FC9797D021D6616600F2FD90 /* BufferToTexture.metal */,
+				FC2BFD3A21DF480300C262B2 /* CPUCompute.h */,
+				FC2BFD3B21DF480400C262B2 /* CPUCompute.mm */,
+				FC2BFD3D21DF5CE800C262B2 /* PreProcessKernel.metal */,
+				FC2BFD2B21DF3FE900C262B2 /* Genet.swift */,
+				FC2BFD2F21DF3FEA00C262B2 /* MobileNet.swift */,
+				FC2BFD2E21DF3FEA00C262B2 /* MobileNetCombined.swift */,
+				FC2BFD2A21DF3FE900C262B2 /* MobilenetSSD_AR.swift */,
+				FC2BFD2C21DF3FE900C262B2 /* MobileNetSSD.swift */,
+				FC2BFD2D21DF3FE900C262B2 /* YoloNet.swift */,
 			);
 			path = Net;
 			sourceTree = "<group>";
@@ -319,16 +383,19 @@
 				FC9797CF21D6506F00F2FD90 /* mingren.jpg in Resources */,
 				FC704C2221D237FC00F98BAB /* combined_mobilenet_params in Resources */,
 				FC704C1921D2375300F98BAB /* super_params in Resources */,
+				FC2BFCBE21DF15D900C262B2 /* 123.jpg in Resources */,
 				FC039B8920E11C560081E9F8 /* Assets.xcassets in Resources */,
 				FC9797C721D609FB00F2FD90 /* synset.txt in Resources */,
 				FC5E03B221DCE8D90016C137 /* mingren_input_data in Resources */,
 				FC704C1A21D2375300F98BAB /* super_model in Resources */,
 				FC039B8720E11C550081E9F8 /* Main.storyboard in Resources */,
 				FC9797C221D608E000F2FD90 /* mobilenet_model in Resources */,
+				FC2BFCC021DF279900C262B2 /* classify-img-output.png in Resources */,
 				FC203FB221CBFDBA00B37166 /* test.jpg in Resources */,
 				FC704C2321D237FC00F98BAB /* combined_mobilenet_model in Resources */,
 				FC9797C321D608E000F2FD90 /* mobilenet_params in Resources */,
 				FC704C2421D237FC00F98BAB /* yolo_params in Resources */,
+				FC2BFCBC21DF0A8600C262B2 /* 00001.jpg in Resources */,
 				FC9797BE21D6045B00F2FD90 /* banana.jpeg in Resources */,
 				FC704C2521D237FC00F98BAB /* yolo_model in Resources */,
 			);
@@ -380,15 +447,24 @@
 			isa = PBXSourcesBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				FC2BFD3221DF3FEA00C262B2 /* MobileNetSSD.swift in Sources */,
+				FC2BFD3C21DF480400C262B2 /* CPUCompute.mm in Sources */,
+				FC2BFD4321DF5E1E00C262B2 /* PaddleMobileGPU.m in Sources */,
 				FC039B8420E11C550081E9F8 /* ViewController.swift in Sources */,
 				FC803BCE214D27930094B8E5 /* VideoCapture.swift in Sources */,
-				FC013928210204A3008100E3 /* PreProcessKernel.metal in Sources */,
 				FCF437E8214B6DDB00943429 /* MultiPredictViewController.swift in Sources */,
+				FC2BFD3021DF3FEA00C262B2 /* MobilenetSSD_AR.swift in Sources */,
+				FC2BFD3321DF3FEA00C262B2 /* YoloNet.swift in Sources */,
+				FC2BFD3421DF3FEA00C262B2 /* MobileNetCombined.swift in Sources */,
 				FCBCCC552122EF5500D94F7E /* MetalHelper.swift in Sources */,
 				FC803BCD214D27930094B8E5 /* FPSCounter.swift in Sources */,
+				FC2BFD3521DF3FEA00C262B2 /* MobileNet.swift in Sources */,
 				C2E67E5E21524E460013F575 /* LoadPointerViewController.m in Sources */,
+				FC2BFD3121DF3FEA00C262B2 /* Genet.swift in Sources */,
 				FC039B8220E11C550081E9F8 /* AppDelegate.swift in Sources */,
-				FC9797D121D6616600F2FD90 /* BufferToTexture.metal in Sources */,
+				FC2BFD4421DF5E1E00C262B2 /* SuperResolutionNet.swift in Sources */,
+				FC2BFD3E21DF5CE800C262B2 /* PreProcessKernel.metal in Sources */,
+				FC2BFD3821DF46DE00C262B2 /* OCDemoViewController.m in Sources */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};

metal/paddle-mobile-demo/paddle-mobile-demo/Base.lproj/Main.storyboard

 <?xml version="1.0" encoding="UTF-8"?>
-<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="14113" targetRuntime="iOS.CocoaTouch" propertyAccessControl="none" useAutolayout="YES" useTraitCollections="YES" useSafeAreas="YES" colorMatched="YES" initialViewController="BYZ-38-t0r">
+<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="14460.31" targetRuntime="iOS.CocoaTouch" propertyAccessControl="none" useAutolayout="YES" useTraitCollections="YES" useSafeAreas="YES" colorMatched="YES" initialViewController="BYZ-38-t0r">
     <device id="retina4_7" orientation="portrait">
         <adaptation id="fullscreen"/>
     </device>
     <dependencies>
         <deployment identifier="iOS"/>
-        <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="14088"/>
-        <capability name="Aspect ratio constraints" minToolsVersion="5.1"/>
+        <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="14460.20"/>
         <capability name="Safe area layout guides" minToolsVersion="9.0"/>
         <capability name="documents saved in the Xcode 8 format" minToolsVersion="8.0"/>
     </dependencies>
@@ -20,7 +19,7 @@
                         <autoresizingMask key="autoresizingMask" widthSizable="YES" heightSizable="YES"/>
                         <subviews>
                             <button opaque="NO" contentMode="scaleToFill" contentHorizontalAlignment="center" contentVerticalAlignment="center" buttonType="roundedRect" lineBreakMode="middleTruncation" translatesAutoresizingMaskIntoConstraints="NO" id="TQt-X9-PdF">
-                                <rect key="frame" x="164" y="318" width="46" height="30"/>
+                                <rect key="frame" x="164.5" y="318.5" width="46" height="30"/>
                                 <state key="normal" title="Button"/>
                                 <connections>
                                     <action selector="predictAct:" destination="Vwd-lt-764" eventType="touchUpInside" id="d4z-Cv-6jY"/>
@@ -60,7 +59,7 @@
                                 <nil key="highlightedColor"/>
                             </label>
                             <pickerView contentMode="scaleToFill" translatesAutoresizingMaskIntoConstraints="NO" id="DlO-dk-RMr">
-                                <rect key="frame" x="55" y="510.5" width="320" height="80"/>
+                                <rect key="frame" x="55" y="510" width="320" height="80"/>
                                 <constraints>
                                     <constraint firstAttribute="height" constant="80" id="Sbi-05-Mwd"/>
                                 </constraints>
@@ -83,6 +82,9 @@
                             <button opaque="NO" contentMode="scaleToFill" contentHorizontalAlignment="center" contentVerticalAlignment="center" buttonType="roundedRect" showsTouchWhenHighlighted="YES" lineBreakMode="middleTruncation" translatesAutoresizingMaskIntoConstraints="NO" id="wUL-9N-u1V">
                                 <rect key="frame" x="16" y="597" width="63.5" height="30"/>
                                 <color key="backgroundColor" white="0.0" alpha="1" colorSpace="custom" customColorSpace="genericGamma22GrayColorSpace"/>
+                                <constraints>
+                                    <constraint firstAttribute="width" secondItem="wUL-9N-u1V" secondAttribute="height" multiplier="21:10" id="cp7-bd-CvU"/>
+                                </constraints>
                                 <state key="normal" title="Image">
                                     <color key="titleColor" white="1" alpha="1" colorSpace="custom" customColorSpace="genericGamma22GrayColorSpace"/>
                                 </state>

metal/paddle-mobile-demo/paddle-mobile-demo/LoadPointerViewController.h

-//
-//  LoadPointerViewController.h
-//  paddle-mobile-demo
-//
-//  Created by Xiao,Haichun on 2018/9/19.
-//  Copyright © 2018年 orange. All rights reserved.
-//
-
-#import <UIKit/UIKit.h>
-
-@interface LoadPointerViewController : UIViewController
-
-@end

metal/paddle-mobile-demo/paddle-mobile-demo/Net/MetalHelper.swift → metal/paddle-mobile-demo/paddle-mobile-demo/MetalHelper.swift

@@ -27,7 +27,4 @@ public class MetalHelper {
     queue = device.makeCommandQueue()!
     textureLoader = MTKTextureLoader.init(device: device)
   }
-  
-
 }
-

metal/paddle-mobile/paddle-mobile/CustomNet/Genet.swift → metal/paddle-mobile-demo/paddle-mobile-demo/Net/Genet.swift

@@ -13,41 +13,35 @@
  limitations under the License. */
 
 import Foundation
+import paddle_mobile
 
 public class Genet: Net {
   @objc public override init(device: MTLDevice) {
     super.init(device: device)
-    means = [128.0, 128.0, 128.0]
-    scale = 0.017
-    except = 0
     modelPath = Bundle.main.path(forResource: "genet_model", ofType: nil) ?! "model null"
     paramPath = Bundle.main.path(forResource: "genet_params", ofType: nil) ?! "para null"
-    modelDir = ""
     preprocessKernel = GenetPreProccess.init(device: device)
-    inputDim_ = Dim.init(inDim: [1, 128, 128, 3])
+    inputDim = Dim.init(inDim: [1, 128, 128, 3])
   }
   
-  @objc override public init(device: MTLDevice,paramPointer: UnsafeMutableRawPointer, paramSize:Int, modePointer: UnsafeMutableRawPointer, modelSize: Int) {
-    super.init(device:device,paramPointer:paramPointer,paramSize:paramSize,modePointer:modePointer,modelSize:modelSize)
-    means = [128.0, 128.0, 128.0]
-    scale = 0.017
-    except = 0
-    modelPath = ""
-    paramPath = ""
-    modelDir = ""
+  @objc override public init(device: MTLDevice, paramPointer: UnsafeMutableRawPointer, paramSize:Int, modePointer: UnsafeMutableRawPointer, modelSize: Int) {
+    super.init(device: device,
+               paramPointer: paramPointer,
+               paramSize: paramSize,
+               modePointer: modePointer,
+               modelSize: modelSize)
     preprocessKernel = GenetPreProccess.init(device: device)
-    inputDim_ = Dim.init(inDim: [1, 128, 128, 3])
+    inputDim = Dim.init(inDim: [1, 128, 128, 3])
   }
 
   class GenetPreProccess: CusomKernel {
     init(device: MTLDevice) {
       let s = Shape.init(inWidth: 128, inHeight: 128, inChannel: 3)
-      super.init(device: device, inFunctionName: "genet_preprocess", outputDim: s, usePaddleMobileLib: false)
+      super.init(device: device, inFunctionName: "genet_preprocess", outputDim: s, metalLoadModel: .LoadMetalInDefaultLib, metalLibPath: nil)
     }
   }
   
   override  public func resultStr(res: ResultHolder) -> String {
-//    fatalError()
     return " \(res.result[0]) ... "
   }
   

metal/paddle-mobile/paddle-mobile/CustomNet/MobileNet.swift → metal/paddle-mobile-demo/paddle-mobile-demo/Net/MobileNet.swift

@@ -13,13 +13,14 @@
  limitations under the License. */
 
 import Foundation
+import paddle_mobile
 
 public class MobileNet: Net{
   
   class MobilenetPreProccess: CusomKernel {
     init(device: MTLDevice) {
       let s = Shape.init(inWidth: 224, inHeight: 224, inChannel: 3)
-      super.init(device: device, inFunctionName: "mobilenet_preprocess", outputDim: s, usePaddleMobileLib: false)
+      super.init(device: device, inFunctionName: "mobilenet_preprocess", outputDim: s, metalLoadModel: .LoadMetalInDefaultLib, metalLibPath: nil)
     }
   }
   
@@ -53,14 +54,13 @@ public class MobileNet: Net{
     
   override public init(device: MTLDevice) {
     super.init(device: device)
-    means = [123.68, 116.78, 103.94]
-    scale = 0.017
     except = 0
     modelPath = Bundle.main.path(forResource: "mobilenet_model", ofType: nil) ?! "model null"
-    paramPath = Bundle.main.path(forResource: "mobilenet_params", ofType: nil) ?! "para null"
-    modelDir = ""
+    paramPath = Bundle.main.path(forResource: "mobilenet_params", ofType: nil) ?! "para null"    
+//    metalLoadMode = .LoadMetalInCustomMetalLib
+//    metalLibPath = Bundle.main.path(forResource: "PaddleMobileMetal", ofType: "metallib") ?! " can't be nil "
     preprocessKernel = MobilenetPreProccess.init(device: device)
-    inputDim_ = Dim.init(inDim: [1, 224, 224, 3])
+    inputDim = Dim.init(inDim: [1, 224, 224, 3])
   }
 }
 

metal/paddle-mobile-demo/paddle-mobile-demo/Net/MobileNetCombined.swift

+/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+ 
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+ 
+ http://www.apache.org/licenses/LICENSE-2.0
+ 
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License. */
+
+import Foundation
+import paddle_mobile
+
+public class MobileNetCombined: Net {
+  @objc public override init(device: MTLDevice) {
+    super.init(device: device)
+    except = 0
+    modelPath = Bundle.main.path(forResource: "combined_mobilenet_model", ofType: nil) ?! "model null"
+    paramPath = Bundle.main.path(forResource: "combined_mobilenet_params", ofType: nil) ?! "para null"
+    inputDim = Dim.init(inDim: [1, 224, 224, 3])
+//    metalLoadMode = .LoadMetalInCustomMetalLib
+//    metalLibPath = Bundle.main.path(forResource: "PaddleMobileMetal", ofType: "metallib") ?! " can't be nil "
+  }
+  
+  override  public func resultStr(res: ResultHolder) -> String {
+    return " \(res.result[0]) ... "
+  }
+  
+}

metal/paddle-mobile/paddle-mobile/CustomNet/MobileNetSSD.swift → metal/paddle-mobile-demo/paddle-mobile-demo/Net/MobileNetSSD.swift

@@ -13,36 +13,35 @@
  limitations under the License. */
 
 import Foundation
+import paddle_mobile
 
-public class MobileNet_ssd_hand: Net{
+public class MobileNet_ssd_hand: Net {
   @objc public override init(device: MTLDevice) {
     super.init(device: device)
-    means = [123.68, 116.78, 103.94]
-    scale = 0.017
     except = 2
     modelPath = Bundle.main.path(forResource: "ssd_hand_model", ofType: nil) ?! "model null"
     paramPath = Bundle.main.path(forResource: "ssd_hand_params", ofType: nil) ?! "para null"
-    modelDir = ""
+//    metalLoadMode = .LoadMetalInCustomMetalLib
+//    metalLibPath = Bundle.main.path(forResource: "PaddleMobileMetal", ofType: "metallib") ?! " can't be nil "
     preprocessKernel = MobilenetssdPreProccess.init(device: device)
-    inputDim_ = Dim.init(inDim: [1, 300, 300, 3])
+    inputDim = Dim.init(inDim: [1, 300, 300, 3])
   }
   
   @objc override public init(device: MTLDevice,paramPointer: UnsafeMutableRawPointer, paramSize:Int, modePointer: UnsafeMutableRawPointer, modelSize: Int) {
     super.init(device:device,paramPointer:paramPointer,paramSize:paramSize,modePointer:modePointer,modelSize:modelSize)
-    means = [123.68, 116.78, 103.94]
-    scale = 0.017
     except = 2
     modelPath = ""
     paramPath = ""
-    modelDir = ""
+//    metalLoadMode = .LoadMetalInCustomMetalLib
+//    metalLibPath = Bundle.main.path(forResource: "PaddleMobileMetal", ofType: "metallib") ?! " can't be nil "
     preprocessKernel = MobilenetssdPreProccess.init(device: device)
-    inputDim_ = Dim.init(inDim: [1, 300, 300, 3])
+    inputDim = Dim.init(inDim: [1, 300, 300, 3])
   }
   
   class MobilenetssdPreProccess: CusomKernel {
     init(device: MTLDevice) {
       let s = Shape.init(inWidth: 300, inHeight: 300, inChannel: 3)
-      super.init(device: device, inFunctionName: "mobilenet_ssd_preprocess", outputDim: s, usePaddleMobileLib: false)
+      super.init(device: device, inFunctionName: "mobilenet_ssd_preprocess", outputDim: s, metalLoadModel: .LoadMetalInDefaultLib, metalLibPath: nil)
     }
   }
   
@@ -50,7 +49,7 @@ public class MobileNet_ssd_hand: Net{
     return " \(res)"
   }
   
-  override func fetchResult(paddleMobileRes: GPUResultHolder) -> ResultHolder {
+  override public func fetchResult(paddleMobileRes: GPUResultHolder) -> ResultHolder {
 
 //    guard let interRes = paddleMobileRes.intermediateResults else {
 //      fatalError(" need have inter result ")

metal/paddle-mobile/paddle-mobile/CustomNet/MobilenetSSD_AR.swift → metal/paddle-mobile-demo/paddle-mobile-demo/Net/MobilenetSSD_AR.swift

@@ -13,36 +13,29 @@
  limitations under the License. */
 
 import Foundation
+import paddle_mobile
 
-public class MobileNet_ssd_AR: Net{
+public class MobileNet_ssd_AR: Net {
   @objc public override init(device: MTLDevice) {
     super.init(device: device)
-    means = [103.94, 116.78, 123.68]
-    scale = 1
     except = 2
     modelPath = Bundle.main.path(forResource: "ar_model", ofType: nil) ?! "model null"
     paramPath = Bundle.main.path(forResource: "ar_params", ofType: nil) ?! "para null"
-    modelDir = ""
     preprocessKernel = MobilenetssdPreProccess.init(device: device)
-    inputDim_ = Dim.init(inDim: [1, 160, 160, 3])
+    inputDim = Dim.init(inDim: [1, 160, 160, 3])
   }
   
   @objc override public init(device: MTLDevice,paramPointer: UnsafeMutableRawPointer, paramSize:Int, modePointer: UnsafeMutableRawPointer, modelSize: Int) {
     super.init(device:device,paramPointer:paramPointer,paramSize:paramSize,modePointer:modePointer,modelSize:modelSize)
-    means = [103.94, 116.78, 123.68]
-    scale = 1
     except = 2
-    modelPath = ""
-    paramPath = ""
-    modelDir = ""
     preprocessKernel = MobilenetssdPreProccess.init(device: device)
-    inputDim_ = Dim.init(inDim: [1, 160, 160, 3])
+    inputDim = Dim.init(inDim: [1, 160, 160, 3])
   }
   
   class MobilenetssdPreProccess: CusomKernel {
-    init(device: MTLDevice) {
+    init(device: MTLDevice)  {
       let s = Shape.init(inWidth: 160, inHeight: 160, inChannel: 3)
-      super.init(device: device, inFunctionName: "mobilent_ar_preprocess", outputDim: s, usePaddleMobileLib: false)
+      super.init(device: device, inFunctionName: "mobilent_ar_preprocess", outputDim: s, metalLoadModel: .LoadMetalInDefaultLib, metalLibPath: nil)
     }
   }
   
@@ -50,18 +43,19 @@ public class MobileNet_ssd_AR: Net{
     return " \(res.result[0])"
   }
   
-  override func fetchResult(paddleMobileRes: GPUResultHolder) -> ResultHolder {
-    guard let interRes = paddleMobileRes.intermediateResults else {
-      fatalError(" need have inter result ")
-    }
-    
-    guard let scores = interRes["Scores"], scores.count > 0, let score = scores[0] as?  FetchHolder else {
-      fatalError(" need score ")
-    }
-    
-    guard let bboxs = interRes["BBoxes"], bboxs.count > 0, let bbox = bboxs[0] as? FetchHolder else {
-      fatalError()
-    }
+  override public func fetchResult(paddleMobileRes: GPUResultHolder) -> ResultHolder {
+    fatalError()
+//    guard let interRes = paddleMobileRes.intermediateResults else {
+//      fatalError(" need have inter result ")
+//    }
+//
+//    guard let scores = interRes["Scores"], scores.count > 0, let score = scores[0] as?  FetchHolder else {
+//      fatalError(" need score ")
+//    }
+//
+//    guard let bboxs = interRes["BBoxes"], bboxs.count > 0, let bbox = bboxs[0] as? FetchHolder else {
+//      fatalError()
+//    }
     
 //    let startDate = Date.init()
     
@@ -72,19 +66,19 @@ public class MobileNet_ssd_AR: Net{
 //
 //    print((0..<bbox.capacity).map{ bbox.result[$0] }.strideArray())
     
-    let nmsCompute = NMSCompute.init()
-    nmsCompute.scoreThredshold = 0.25
-    nmsCompute.nmsTopK = 100
-    nmsCompute.keepTopK = 100
-    nmsCompute.nmsEta = 1.0
-    nmsCompute.nmsThreshold = 0.449999988
-    nmsCompute.background_label = 0;
-    nmsCompute.scoreDim = [NSNumber.init(value: score.dim[0]), NSNumber.init(value: score.dim[1]), NSNumber.init(value: score.dim[2])]
-    nmsCompute.bboxDim = [NSNumber.init(value: bbox.dim[0]), NSNumber.init(value: bbox.dim[1]), NSNumber.init(value: bbox.dim[2])]
-    guard let result = nmsCompute.compute(withScore: score.result, andBBoxs: bbox.result) else {
-      fatalError( " result error " )
-    }
-    let resultHolder = ResultHolder.init(inResult: result.output, inCapacity: Int(result.outputSize))
+//    let nmsCompute = NMSCompute.init()
+//    nmsCompute.scoreThredshold = 0.25
+//    nmsCompute.nmsTopK = 100
+//    nmsCompute.keepTopK = 100
+//    nmsCompute.nmsEta = 1.0
+//    nmsCompute.nmsThreshold = 0.449999988
+//    nmsCompute.background_label = 0;
+//    nmsCompute.scoreDim = [NSNumber.init(value: score.dim[0]), NSNumber.init(value: score.dim[1]), NSNumber.init(value: score.dim[2])]
+//    nmsCompute.bboxDim = [NSNumber.init(value: bbox.dim[0]), NSNumber.init(value: bbox.dim[1]), NSNumber.init(value: bbox.dim[2])]
+//    guard let result = nmsCompute.compute(withScore: score.result, andBBoxs: bbox.result) else {
+//      fatalError( " result error " )
+//    }
+//    let resultHolder = ResultHolder.init(inResult: result.output, inCapacity: Int(result.outputSize))
 //    for i in 0..<Int(result.outputSize) {
 //
 //      print("i \(i) : \(result.output[i])")
@@ -92,10 +86,11 @@ public class MobileNet_ssd_AR: Net{
 //    print(Date.init().timeIntervalSince(startDate))
 
 //    print(resultHolder.result![0])
-    return resultHolder
+//    return resultHolder
   }
   
-  override func updateProgram(program: Program) {
+//  override func updateProgram(program: Program) {
+  
 //    for i in [56, 66, 76, 86, 93, 99] {
 //      let opDesc = program.programDesc.blocks[0].ops[i]
 //      let output = opDesc.outputs["Out"]!.first!
@@ -148,6 +143,6 @@ public class MobileNet_ssd_AR: Net{
 //      print(" split axis \(opDesc.attrs["axis"])")
 //    }
     // 99
-  }
+//  }
   
 }

metal/paddle-mobile-demo/paddle-mobile-demo/Net/PaddleMobile.swift

-//
-//  PaddleMobile.swift
-//  paddle-mobile-demo
-//
-//  Created by liuRuiLong on 2018/9/5.
-//  Copyright © 2018年 orange. All rights reserved.
-//
-
-import Foundation

metal/paddle-mobile-demo/paddle-mobile-demo/Net/PreProcessKernel.metal

@@ -115,23 +115,3 @@ kernel void mobilent_ar_preprocess_half(texture2d<half, access::read> inTexture
   const half4 inColor = (inTexture.read(gid) * 255.0 - means) * 0.017;
   outTexture.write(half4(inColor.z, inColor.y, inColor.x, 0.0f), gid);
 }
-
-kernel void scale(texture2d<float, access::sample> inTexture [[texture(0)]], texture2d<float, access::write> outTexture [[texture(1)]], uint2 gid [[thread_position_in_grid]]) {
-  if (gid.x >= outTexture.get_width() ||
-      gid.y >= outTexture.get_height()) return;
-  float w_stride = inTexture.get_width() / outTexture.get_width();
-  float h_stride = inTexture.get_height() / outTexture.get_height();
-  constexpr sampler sample(coord::pixel, filter::nearest, address::clamp_to_zero);
-  float4 input = inTexture.sample(sample, float2(gid.x * w_stride,    gid.y * h_stride), 0);
-  outTexture.write(input, gid);
-}
-
-kernel void scale_half(texture2d<float, access::sample> inTexture [[texture(0)]], texture2d<half, access::write> outTexture [[texture(1)]], uint2 gid [[thread_position_in_grid]]) {
-  if (gid.x >= outTexture.get_width() ||
-      gid.y >= outTexture.get_height()) return;
-  float w_stride = inTexture.get_width() / outTexture.get_width();
-  float h_stride = inTexture.get_height() / outTexture.get_height();
-  constexpr sampler sample(coord::pixel, filter::nearest, address::clamp_to_zero);
-  float4 input = inTexture.sample(sample, float2(gid.x * w_stride,    gid.y * h_stride), 0);
-  outTexture.write(half4(input), gid);
-}

metal/paddle-mobile-demo/paddle-mobile-demo/Net/YoloNet.swift

+/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+ 
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+ 
+ http://www.apache.org/licenses/LICENSE-2.0
+ 
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License. */
+
+import Metal
+import Foundation
+import paddle_mobile
+
+public class YoloNet: Net {
+  @objc public override init(device: MTLDevice) {
+    super.init(device: device)
+    except = 0
+    modelPath = Bundle.main.path(forResource: "yolo_model", ofType: nil) ?! "model null"
+    paramPath = Bundle.main.path(forResource: "yolo_params", ofType: nil) ?! "para null"
+    inputDim = Dim.init(inDim: [1, 416, 416, 3])
+//    metalLoadMode = .LoadMetalInCustomMetalLib
+//    metalLibPath = Bundle.main.path(forResource: "PaddleMobileMetal", ofType: "metallib") ?! " can't be nil "
+  }
+
+  override  public func resultStr(res: ResultHolder) -> String {
+    return " \(res.result[0]) ... "
+  }
+  
+}

metal/paddle-mobile-demo/paddle-mobile-demo/OCDemo/LoadPointerViewController.h

+/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+ 
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+ 
+ http://www.apache.org/licenses/LICENSE-2.0
+ 
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License. */
+
+#import <UIKit/UIKit.h>
+
+@interface LoadPointerViewController : UIViewController
+
+@end

metal/paddle-mobile-demo/paddle-mobile-demo/LoadPointerViewController.m → metal/paddle-mobile-demo/paddle-mobile-demo/OCDemo/LoadPointerViewController.m

-//
-//  LoadPointerViewController.m
-//  paddle-mobile-demo
-//
-//  Created by Xiao,Haichun on 2018/9/19.
-//  Copyright © 2018年 orange. All rights reserved.
-//
+/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+ 
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+ 
+ http://www.apache.org/licenses/LICENSE-2.0
+ 
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License. */
 
+#import "PaddleMobileGPU.h"
 #import "LoadPointerViewController.h"
-#import <Metal/Metal.h>
 #import "paddle-mobile-demo-Bridging-Header.h"
 
+#import <Metal/Metal.h>
+
 @interface LoadPointerViewController ()
 
 @property (strong, nonatomic) id<MTLDevice> device;

metal/paddle-mobile-demo/paddle-mobile-demo/OCDemo/OCDemoViewController.h

+/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+ 
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+ 
+ http://www.apache.org/licenses/LICENSE-2.0
+ 
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License. */
+#import <Foundation/Foundation.h>
+
+NS_ASSUME_NONNULL_BEGIN
+
+@interface OCDemoViewController : NSObject
+
+@end
+
+NS_ASSUME_NONNULL_END

metal/paddle-mobile-demo/paddle-mobile-demo/OCDemo/OCDemoViewController.m

+/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+ 
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+ 
+ http://www.apache.org/licenses/LICENSE-2.0
+ 
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License. */
+
+#import "OCDemoViewController.h"
+
+@implementation OCDemoViewController
+
+@end

metal/paddle-mobile/paddle-mobile/PaddleMobileGPU.h → metal/paddle-mobile-demo/paddle-mobile-demo/OCInterface/PaddleMobileGPU.h

@@ -16,9 +16,8 @@
 #import <Foundation/Foundation.h>
 
 typedef enum : NSUInteger {
-  MobileNetType,
-  MobileNetSSDType,
-  GenetType,
+  SuperResolutionNetType,
+  MobileNetSSDType
 } NetType;
 
 @interface PaddleMobileGPUResult: NSObject

metal/paddle-mobile/paddle-mobile/PaddleMobileGPU.m → metal/paddle-mobile-demo/paddle-mobile-demo/OCInterface/PaddleMobileGPU.m

@@ -12,11 +12,10 @@
  See the License for the specific language governing permissions and
  limitations under the License. */
 
-#import "paddle_mobile.h"
 #import "PaddleMobileGPU.h"
 
 #import <Foundation/Foundation.h>
-#import <paddle_mobile/paddle_mobile-Swift.h>
+#import <paddle_mobile_demo-Swift.h>
 
 @implementation ModelConfig
 @end
@@ -53,12 +52,10 @@
   self = [super init];
   if (self) {
     Net *net = nil;
-    if (netType == GenetType) {
-      net = [[Genet alloc] initWithDevice:queue.device paramPointer:config.paramPointer paramSize:config.paramSize modePointer:config.modelPointer modelSize:config.modelSize];
+    if (netType == SuperResolutionNetType) {
+      net = [[SuperResolutionNet alloc] initWithDevice:queue.device];
     } else if (netType == MobileNetSSDType) {
       net = [[MobileNet_ssd_AR alloc] initWithDevice:queue.device paramPointer:config.paramPointer paramSize:config.paramSize modePointer:config.modelPointer modelSize:config.modelSize];
-    } else if (netType == MobileNetType) {
-      
     }
     runner = [[Runner alloc] initInNet:net commandQueue:queue];
   }

metal/paddle-mobile/paddle-mobile/CustomNet/SuperResolutionNet.swift → metal/paddle-mobile-demo/paddle-mobile-demo/OCInterface/SuperResolutionNet.swift

@@ -13,31 +13,25 @@
  limitations under the License. */
 
 import Foundation
+import paddle_mobile
 
-
-public class SuperResolutionNet: Net{
+@objc public class SuperResolutionNet: Net{
   override public func resultStr(res: ResultHolder) -> String {
     return "未实现"
   }
   
-  override public init(device: MTLDevice) {
+  @objc override public init(device: MTLDevice) {
     super.init(device: device)
-    means = [0.0, 0.0, 0.0]
-    scale = 1.0
     except = 0
     modelPath = Bundle.main.path(forResource: "super_model", ofType: nil) ?! "model null"
     paramPath = Bundle.main.path(forResource: "super_params", ofType: nil) ?! "para null"
-    modelDir = ""
     preprocessKernel = nil
-//    inputDim_ = Dim.init(inDim: [1, Int(552 * 1.414), Int(310 * 1.414), 1])
-    inputDim_ = Dim.init(inDim: [1, 224, 224, 1])
+    inputDim = Dim.init(inDim: [1, 224, 224, 1])
+//    metalLoadMode = .LoadMetalInCustomMetalLib
+//    metalLibPath = Bundle.main.path(forResource: "PaddleMobileMetal", ofType: "metallib") ?! " can't be nil "
   }
   
-  override func updateProgram(program: Program) {
-    guard needUpdateProgram else {
-      return
-    }
-    
+  override public func updateProgram(program: Program) {
     // n h w c
     for block in program.programDesc.blocks {
       for varDesc in block.vars {
@@ -47,8 +41,9 @@ public class SuperResolutionNet: Net{
             if let texture = varEle as? Texture {
               let newDim = Dim.init(inDim: [texture.dim[0],  inputDim[1], inputDim[2], texture.tensorDim[1]])
               print(" var desc name " + varDesc.name + " new dim" + "\(newDim)")
+              
               texture.updateDims(inTensorDim: Dim.init(inDim: [texture.tensorDim[0], texture.tensorDim[1], inputDim[1], inputDim[2]]), inDim: newDim)
-              texture.initTexture(device: device, inTranspose: [0, 1, 2, 3], computePrecision: computePrecision)
+              texture.initTexture(device: device, inTranspose: [0, 1, 2, 3], computePrecision: GlobalConfig.shared.computePrecision)
               
               let output: FetchHolder = program.scope.output() as! FetchHolder
               output.dim = newDim
@@ -60,7 +55,6 @@ public class SuperResolutionNet: Net{
         }
       }
     }
-    needUpdateProgram = false
   }
 }
 

metal/paddle-mobile-demo/paddle-mobile-demo/ViewController.swift

@@ -18,7 +18,6 @@ import CoreMedia
 import paddle_mobile
 import MetalPerformanceShaders
 
-
 class FileReader {
   let file: UnsafeMutablePointer<FILE>
   let fileSize: Int
@@ -53,16 +52,12 @@ enum Platform {
 let platformSupport: [(Platform, String)] = [(.GPU, "GPU")]
 
 enum SupportModel: String{
-  //  case mobilenet = "mobilenet"
-  //  case mobilenet_ssd    = "mobilenetssd"
   case yolo               = "yolo"
   case mobilenet_combined = "mobilenet_combined"
   case super_resolution   = "superresoltion"
   case mobilenet          = "mobilenet"
   
   static func supportedModels() -> [SupportModel] {
-    // .mobilenet,
-    // .mobilenet_ssd,
     return [.super_resolution, .yolo, .mobilenet_combined, .mobilenet]
   }
 }
@@ -94,24 +89,25 @@ class ViewController: UIViewController {
   @IBAction func loadAct(_ sender: Any) {
     runner = Runner.init(inNet: netSupport[modelType]!, commandQueue: MetalHelper.shared.queue)
     if platform == .GPU {
-      let filePath = Bundle.main.path(forResource: "mingren_input_data", ofType: nil)
-      let fileReader = try! FileReader.init(paramPath: filePath!)
-      let pointer: UnsafeMutablePointer<Float32> = fileReader.read()
-      
-      
-      let buffer = MetalHelper.shared.device.makeBuffer(length: fileReader.fileSize, options: .storageModeShared)
-      
-      buffer?.contents().copyMemory(from: pointer, byteCount: fileReader.fileSize)
+//      let filePath = Bundle.main.path(forResource: "mingren_input_data", ofType: nil)
+//      let fileReader = try! FileReader.init(paramPath: filePath!)
+//      let pointer: UnsafeMutablePointer<Float32> = fileReader.read()
+//      
+//      
+//      let buffer = MetalHelper.shared.device.makeBuffer(length: fileReader.fileSize, options: .storageModeShared)
+//      
+//      buffer?.contents().copyMemory(from: pointer, byteCount: fileReader.fileSize)
       
       
       if self.toPredictTexture == nil {
         
-        runner.getTexture(inBuffer: buffer!) { [weak self] (texture) in
+//        runner.getTexture(inBuffer: buffer!) { [weak self] (texture) in
+//          self?.toPredictTexture = texture
+//        }
+        
+        runner.getTexture(image: selectImage!.cgImage!) { [weak self] (texture) in
           self?.toPredictTexture = texture
         }
-        
-//        runner.getTexture(image: selectImage!.cgImage!) { [weak self] (texture) in
-//        }
       }
     } else {
       fatalError( " unsupport " )
@@ -178,13 +174,14 @@ class ViewController: UIViewController {
     modelPickerView.dataSource = self
     threadPickerView.delegate = self
     threadPickerView.dataSource = self
-    if let image = UIImage.init(named: "test.jpg") {
+    if let image = UIImage.init(named: "classify-img-output.png") {
       selectImage = image
       selectImageView.image = image
     } else {
       print("请添加测试图片")
     }
     
+    GlobalConfig.shared.computePrecision = .Float32
     
     //    if platform == .CPU {
     //      inputPointer = runner.preproccess(image: selectImage!.cgImage!)

metal/paddle-mobile/paddle-mobile/API/GlobalConfig.swift

+/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+ 
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+ 
+ http://www.apache.org/licenses/LICENSE-2.0
+ 
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License. */
+
+import Foundation
+
+@objc public enum MetalLoadMode: Int {
+  case
+  LoadMetalInPaddleMobile   = 1,     // 使用 paddle-mobile 中的 metal 代码
+  LoadMetalInDefaultLib     = 2,     // 使用 main bundle 中的 metal 代码
+  LoadMetalInCustomMetalLib = 3      // 使用 metal 库文件
+}
+
+@objc public enum ComputePrecision: Int {
+  case
+  Float32 = 1,
+  Float16 = 2
+}
+
+@objc public class GlobalConfig: NSObject {
+  
+  /// 单例
+  @objc public static let shared: GlobalConfig = GlobalConfig.init()
+  
+  /// 运算精度， runner 生命周期中不可变
+  @objc public var computePrecision: ComputePrecision = .Float16
+
+}

metal/paddle-mobile/paddle-mobile/Net.swift → metal/paddle-mobile/paddle-mobile/API/Net.swift

@@ -12,58 +12,52 @@
  See the License for the specific language governing permissions and
  limitations under the License. */
 
-
-import Foundation
 import Metal
+import Foundation
 
-public class ResultHolder: NSObject {
-  @objc public let result: UnsafeMutablePointer<Float32>
-  @objc public let capacity: Int
+/// 网络的基类， 参数已经给了默认值，请在子类实现中修改需要改的参数
+@objc open class Net: NSObject {
 
-  init(inResult: UnsafeMutablePointer<Float32>, inCapacity: Int) {
-    result = inResult
-    capacity = inCapacity
-  }
+  /// 默认为0， 如果指定个数， 后边 except 个op不使用 GPU 运算， 中间结果会通过 fetchResult 传参过来
+  @objc public var except: Int = 0
+      
+  /// 预处理 kernel， 如果输入图像需要预处理， 则指定预处理 kernel
+  @objc public var preprocessKernel: CusomKernel? = nil
   
-  @objc public func releasePointer() {
-    result.deinitialize(count: capacity)
-    result.deallocate()
-  }
-}
-
-public class Net: NSObject {
-  var except: Int = 0
+  // 以下四个参数为从内存中读取模型时用到的参数
+  /// 模型在内存中的指针
+  @objc public var modelPointer: UnsafeMutableRawPointer? = nil
   
-  // for CPU
-  var means: [Float] = []
-  var scale: Float = 0.0
+  /// 模型大小 单位： 字节
+  @objc public var modelSize: Int = 0
   
-  var needUpdateProgram = true
+  /// 权重参数在内存中的指针
+  @objc public var paramPointer: UnsafeMutableRawPointer? = nil
   
-  public var inputDim: Dim {
-    get{
-      return inputDim_
-    }
-    set{
-      if inputDim_ != newValue {
-        needUpdateProgram = true
-      }
-      inputDim_ = newValue
-    }
-  }
+  /// 权重大小 单位： 字节
+  @objc public var paramSize: Int = 0
   
-  var inputDim_: Dim = Dim.init(inDim: [])
-  var preprocessKernel: CusomKernel? = nil
-  var paramPointer: UnsafeMutableRawPointer? = nil
-  var paramSize: Int = 0
-  var modelPointer: UnsafeMutableRawPointer? = nil
-  var modelSize: Int = 0
-  var modelPath: String = ""
-  var paramPath: String = ""
-  var modelDir: String = ""
-  let device: MTLDevice
-  
-  @objc public init(device: MTLDevice,paramPointer: UnsafeMutableRawPointer, paramSize:Int, modePointer: UnsafeMutableRawPointer, modelSize: Int) {
+  // 以下两个为从文件中读取模型时用到的参数
+  /// 模型文件路径
+  @objc public var modelPath: String? = nil
+  
+  /// 权重文件路径
+  @objc public var paramPath: String? = nil
+  
+  /// 代表着 GPU 处理器
+  @objc public let device: MTLDevice
+  
+  /// metal 代码加载方式 注意： 如果静态库只能使用 LoadMetalInDefaultLib LoadMetalInCustomMetalLib 进行 load metal 代码
+  @objc public var metalLoadMode: MetalLoadMode = .LoadMetalInPaddleMobile
+  
+  /// 当 metalLoadMode 为 LoadMetalInCustomMetalLib 时， metal library 路径不能为空
+  @objc public var metalLibPath: String? = nil
+  
+  /// 输入维度，按照 n h w c 方式传入
+  @objc public var inputDim: Dim = Dim.init(inDim: [])
+  
+  
+  @objc public init(device: MTLDevice, paramPointer: UnsafeMutableRawPointer, paramSize:Int, modePointer: UnsafeMutableRawPointer, modelSize: Int) {
     self.paramPointer = paramPointer
     self.paramSize = paramSize
     self.modelPointer = modePointer
@@ -76,23 +70,18 @@ public class Net: NSObject {
     self.device = device
     super.init()
   }
-
-  @objc public func updateInputDim(inDim: [Int]) {
-    inputDim = Dim.init(inDim: inDim)
-  }
   
-  public func resultStr(res: ResultHolder) -> String {
+  @objc open func resultStr(res: ResultHolder) -> String {
     fatalError()
   }
   
-  func fetchResult(paddleMobileRes: GPUResultHolder) -> ResultHolder {
+  @objc open func fetchResult(paddleMobileRes: GPUResultHolder) -> ResultHolder {
     guard let inResPointer = paddleMobileRes.resultPointer else {
       fatalError()
     }
-    
     return ResultHolder.init(inResult: inResPointer, inCapacity: paddleMobileRes.capacity)
   }
   
-  func updateProgram(program: Program) {
+  open func updateProgram(program: Program) {
   }
 }

metal/paddle-mobile/paddle-mobile/Runner.swift → metal/paddle-mobile/paddle-mobile/API/Runner.swift

-//
-//  Runner.swift
-//  paddle-mobile
-//
-//  Created by liuRuiLong on 2018/12/27.
-//  Copyright © 2018 orange. All rights reserved.
-//
+/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+ 
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+ 
+ http://www.apache.org/licenses/LICENSE-2.0
+ 
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License. */
 
 import MetalKit
 import Foundation
 
-class ScaleKernel: CusomKernel {
-  init(device: MTLDevice, shape: Shape) {
-    if computePrecision == .Float32 {
-      super.init(device: device, inFunctionName: "scale", outputDim: shape, usePaddleMobileLib: false)
-    } else if computePrecision == .Float16 {
-      super.init(device: device, inFunctionName: "scale_half", outputDim: shape, usePaddleMobileLib: false)
-    } else {
-      fatalError(" unsupport ")
-    }
+@objc public class ResultHolder: NSObject {
+  @objc public let result: UnsafeMutablePointer<Float32>
+  @objc public let capacity: Int
+  
+  init(inResult: UnsafeMutablePointer<Float32>, inCapacity: Int) {
+    result = inResult
+    capacity = inCapacity
+  }
+  
+  @objc public func releasePointer() {
+    result.deinitialize(count: capacity)
+    result.deallocate()
   }
 }
 
@@ -29,15 +38,12 @@ class ScaleKernel: CusomKernel {
   public let net: Net
   let device: MTLDevice?
   let numel: Int
-  let meansNumber: [NSNumber]
   
-  // dims num nchw
-  let dimsNum: [NSNumber]
-  /**
-   * inNet:        需要运行的网络
-   * commandQueue: GPU 是需要传入
-   * inPlatform:   需要使用的平台, GPU or CPU
-   */
+  /// 初始化函数
+  ///
+  /// - Parameters:
+  ///   - inNet: 传入自定义的网络
+  ///   - commandQueue: commandQueue
   @objc public init(inNet: Net, commandQueue: MTLCommandQueue?) {
     guard inNet.inputDim.cout() == 4 else {
       fatalError(" input dim count must 4 ")
@@ -49,21 +55,12 @@ class ScaleKernel: CusomKernel {
     if let inDevice = device {
       textureLoader = MTKTextureLoader.init(device: inDevice)
     }
-
     numel = net.inputDim.numel()
-    meansNumber = net.means.map {
-      NSNumber.init(value: $0)
-    }
-    
-    dimsNum = [NSNumber.init(value: net.inputDim[0]),
-               NSNumber.init(value: net.inputDim[3]),
-               NSNumber.init(value: net.inputDim[1]),
-               NSNumber.init(value: net.inputDim[2])]
   }
   
-  /**
-   * load 模型, 返回 true 可进行预测
-   */
+  /// load 模型, 返回 true 可进行预测
+  ///
+  /// - Returns: load 成功或失败
   @objc public func load() -> Bool {
       guard let inDevice = device, let inQueue = queue else {
         print(" paddle mobile gpu load error, need MTLCommandQueue")
@@ -71,10 +68,24 @@ class ScaleKernel: CusomKernel {
       }
       let loader = Loader<Float32>.init()
       do {
-        //        program = try loader.load(device: inDevice, paramPointer: net.paramPointer!, paramSize: net.paramSize,modePointer:net.modelPointer!,modelSize:net.modelSize)
-        program = try loader.load(device: inDevice, modelPath: net.modelPath, paraPath: net.paramPath)
         
-        executor = try Executor<Float32>.init(inDevice: inDevice, inQueue: inQueue, inProgram: program!)
+        if let inParamPointer = net.paramPointer, let inModelPointer = net.modelPointer {
+          guard net.paramSize > 0 && net.modelSize > 0 else {
+            print(" load from memory param size or model size can't 0 ")
+            return false
+          }
+          program = try loader.load(device: inDevice, paramPointer: inParamPointer, paramSize: net.paramSize,modePointer:inModelPointer,modelSize:net.modelSize)
+        } else if let inModelPath = net.modelPath, let inParamPath = net.paramPath {
+          program = try loader.load(device: inDevice, modelPath: inModelPath, paraPath: inParamPath)
+        } else {
+          print(" model pointer or model file path need be specified")
+          return false
+        }
+        
+        let initContext: InitContext = InitContext.init()
+        initContext.metalLoadMode = net.metalLoadMode
+        initContext.metalLibPath = net.metalLibPath
+        executor = try Executor<Float32>.init(inDevice: inDevice, inQueue: inQueue, inProgram: program!, initContext: initContext)
         net.updateProgram(program: program!)
       } catch let error {
         print(error)
@@ -83,13 +94,12 @@ class ScaleKernel: CusomKernel {
     return true
   }
   
-  /**
-   * GPU 版本 predict
-   * texture: 需要预测的 texture 需要做过预处理
-   * ( _ success: Bool, _ time:TimeInterval, _ resultArray: [Float32]) -> Void : 回调闭包, 三个参数分别为: 是否成功, 预测耗时, 结果数组
-   */
+  /// 预测
+  ///
+  /// - Parameters:
+  ///   - texture: 输入 texture 需要使用 getTexture 获得
+  ///   - completion: 结果回调， 当 success 为 true 时 result 不为 nil
   @objc public func predict(texture: MTLTexture, completion: @escaping ( _ success: Bool, _ result: ResultHolder?) -> Void) {
-    net.updateProgram(program: program!)
     do {
       try self.executor?.predict(input: texture, dim: self.net.inputDim, completionHandle: { [weak self] (res) in
         guard let SSelf = self else {
@@ -105,23 +115,28 @@ class ScaleKernel: CusomKernel {
     }
   }
   
-  /*
-   * 清理内存, 调用此函数后, 不能再使用, 需重新 load
-   */
+  /// 清理内存, 调用此函数后, 不能再使用, 需重新 load
   @objc public func clear() {
     executor?.clear()
     executor = nil
     program = nil
   }
-
-  /*
-   * 获取 texture, 对 texture 进行预处理, GPU 预测时使用
-   */
+  
+  /// 获取 texture, 对 texture 进行预处理, 预测时使用
+  ///
+  /// - Parameters:
+  ///   - image: 输入图像
+  ///   - getTexture: 获取 texture 回调
   @objc public func getTexture(image: CGImage, getTexture: @escaping (MTLTexture) -> Void) {
     let texture = try? textureLoader?.newTexture(cgImage: image, options: [:]) ?! " texture loader error"
     scaleTexture(input: texture!, complete: getTexture)
   }
   
+  /// 通过 buffer 获取 texture， 内部会使用GPU进行转换操作
+  ///
+  /// - Parameters:
+  ///   - inBuffer: 输入buffer
+  ///   - getTexture: 结果回调
   @objc public func getTexture(inBuffer: MTLBuffer, getTexture: @escaping (MTLTexture) -> Void) {
     guard let inQueue = queue, let inDevice = device else {
       fatalError( " queue or devcie nil " )
@@ -131,7 +146,7 @@ class ScaleKernel: CusomKernel {
       fatalError( " make buffer error" )
     }
     
-    let bufferToTextureKernel = BufferToTextureKernel.init(device: inDevice, outputDim: Shape.init(inWidth: net.inputDim[2], inHeight: net.inputDim[1], inChannel: net.inputDim[3]))
+    let bufferToTextureKernel = BufferToTextureKernel.init(device: inDevice, outputDim: Shape.init(inWidth: net.inputDim[2], inHeight: net.inputDim[1], inChannel: net.inputDim[3]), metalLoadMode: net.metalLoadMode, metalLibPath: net.metalLibPath)
     do {
       try bufferToTextureKernel.compute(inputBuffer: inBuffer, commandBuffer: buffer)
     } catch {
@@ -144,6 +159,19 @@ class ScaleKernel: CusomKernel {
     
     buffer.commit()
   }
+
+  /// 更新输入维度， 针对可变长输入模型
+  ///
+  /// - Parameter inDim: 输入维度
+  @objc public func updateInputDim(inDim: Dim) {
+    if net.inputDim != inDim {
+      guard let inProgram = program else {
+        fatalError(" need load first ")
+      }
+      net.inputDim = inDim
+      net.updateProgram(program: inProgram)
+    }
+  }
   
   public func scaleTexture(input: MTLTexture , complete: @escaping (MTLTexture) -> Void) {
     
@@ -155,7 +183,7 @@ class ScaleKernel: CusomKernel {
       fatalError( " make buffer error" )
     }
     
-    let scaleKernel = ScaleKernel.init(device: inDevice, shape: Shape.init(inWidth: net.inputDim[2], inHeight: net.inputDim[1], inChannel: 3))
+    let scaleKernel = ScaleKernel.init(device: inDevice, shape: Shape.init(inWidth: net.inputDim[2], inHeight: net.inputDim[1], inChannel: 3), metalLoadMode: net.metalLoadMode, metalLibPath: net.metalLibPath)
     
     do {
       try scaleKernel.compute(inputTexuture: input, commandBuffer: buffer)

metal/paddle-mobile/paddle-mobile/CustomNet/MobileNetCombined.swift

-//
-//  MobileNetCombined.swift
-//  paddle-mobile
-//
-//  Created by Xiao,Haichun on 2018/12/5.
-//  Copyright © 2018 orange. All rights reserved.
-//
-
-import Foundation
-
-public class MobileNetCombined: Net {
-  @objc public override init(device: MTLDevice) {
-    super.init(device: device)
-    means = [0, 0, 0]
-    scale = 1
-    except = 0
-    modelPath = Bundle.main.path(forResource: "combined_mobilenet_model", ofType: nil) ?! "model null"
-    paramPath = Bundle.main.path(forResource: "combined_mobilenet_params", ofType: nil) ?! "para null"
-    modelDir = ""
-    inputDim_ = Dim.init(inDim: [1, 224, 224, 3])
-  }
-  
-  @objc override public init(device: MTLDevice,paramPointer: UnsafeMutableRawPointer, paramSize:Int, modePointer: UnsafeMutableRawPointer, modelSize: Int) {
-
-    super.init(device:device,paramPointer:paramPointer,paramSize:paramSize,modePointer:modePointer,modelSize:modelSize)
-    means = [0, 0, 0]
-    scale = 1
-    except = 0
-    modelPath = ""
-    paramPath = ""
-    modelDir = ""
-    inputDim_ = Dim.init(inDim: [1,  224, 224, 3])
-  }
-  
-  //    class GenetPreProccess: CusomKernel {
-  //        init(device: MTLDevice) {
-  //            let s = CusomKernel.Shape.init(inWidth: 128, inHeight: 128, inChannel: 3)
-  //            super.init(device: device, inFunctionName: "genet_preprocess", outputDim: s, usePaddleMobileLib: false)
-  //        }
-  //    }
-  
-  override  public func resultStr(res: ResultHolder) -> String {
-    //    fatalError()
-    return " \(res.result[0]) ... "
-  }
-  
-}

metal/paddle-mobile/paddle-mobile/CustomNet/YoloNet.swift

-//
-//  YoloNet.swift
-//  paddle-mobile
-//
-//  Created by Xiao,Haichun on 2018/12/5.
-//  Copyright © 2018 orange. All rights reserved.
-//
-
-import Foundation
-import Metal
-
-public class YoloNet: Net {
-  @objc public override init(device: MTLDevice) {
-    super.init(device: device)
-    means = [0, 0, 0]
-    scale = 1
-    except = 0
-    modelPath = Bundle.main.path(forResource: "yolo_model", ofType: nil) ?! "model null"
-    paramPath = Bundle.main.path(forResource: "yolo_params", ofType: nil) ?! "para null"
-    modelDir = ""
-//    preprocessKernel = GenetPreProccess.init(device: device)
-    inputDim_ = Dim.init(inDim: [1, 416, 416, 3])
-    
-  }
-  
-  @objc override public init(device: MTLDevice,paramPointer: UnsafeMutableRawPointer, paramSize:Int, modePointer: UnsafeMutableRawPointer, modelSize: Int) {
-    super.init(device:device,paramPointer:paramPointer,paramSize:paramSize,modePointer:modePointer,modelSize:modelSize)
-    means = [0, 0, 0]
-    scale = 1
-    except = 0
-    modelPath = ""
-    paramPath = ""
-    modelDir = ""
-    //preprocessKernel = GenetPreProccess.init(device: device)
-    inputDim_ = Dim.init(inDim: [1, 416, 416, 3])
-  }
-  
-  //    class GenetPreProccess: CusomKernel {
-  //        init(device: MTLDevice) {
-  //            let s = CusomKernel.Shape.init(inWidth: 128, inHeight: 128, inChannel: 3)
-  //            super.init(device: device, inFunctionName: "genet_preprocess", outputDim: s, usePaddleMobileLib: false)
-  //        }
-  //    }
-  
-  override  public func resultStr(res: ResultHolder) -> String {
-    return " \(res.result[0]) ... "
-  }
-  
-}

metal/paddle-mobile/paddle-mobile/Common/MetalExtension.swift → metal/paddle-mobile/paddle-mobile/Src/Common/MetalExtension.swift

@@ -18,6 +18,7 @@ import CoreMedia
 
 fileprivate var defaultMetalLibrary: MTLLibrary?
 fileprivate var paddleMobileMetalLibrary: MTLLibrary?
+fileprivate var customMetalLibrary: MTLLibrary?
 
 extension MTLDevice {
   func defaultLibrary() -> MTLLibrary {
@@ -31,6 +32,22 @@ extension MTLDevice {
     }
   }
   
+  func customLibrary(metalLibPath: String) -> MTLLibrary {
+    if customMetalLibrary == nil {
+      do {
+        customMetalLibrary = try makeLibrary(filepath: metalLibPath)
+      } catch  let error {
+        fatalError("\(error)")
+      }
+    }
+    
+    if let inMetalLib = customMetalLibrary {
+      return inMetalLib
+    } else {
+      fatalError(" customlib is nil ")
+    }
+  }
+  
   func paddleMobileLibrary() -> MTLLibrary {
     if paddleMobileMetalLibrary == nil {
       guard let path = Bundle.init(for: Kernel.self).path(forResource: "default", ofType: "metallib") else {
@@ -50,8 +67,19 @@ extension MTLDevice {
     }
   }
   
-  func pipeLine(funcName: String, inPaddleMobileLib: Bool = true) -> MTLComputePipelineState {
-    let useLib = inPaddleMobileLib ? paddleMobileLibrary() : defaultLibrary()
+  func pipeLine(funcName: String, metalLoadMode: MetalLoadMode, metalLibPath: String?) -> MTLComputePipelineState {
+    let useLib: MTLLibrary
+    switch metalLoadMode {
+    case .LoadMetalInDefaultLib:
+      useLib = defaultLibrary()
+    case .LoadMetalInPaddleMobile:
+      useLib = paddleMobileLibrary()
+    case .LoadMetalInCustomMetalLib:
+      useLib = customLibrary(metalLibPath: metalLibPath ?! " can't be nil ")
+    default:
+      fatalError()
+    }
+    
     guard let function = useLib.makeFunction(name: funcName) else {
       fatalError(" function " + funcName + " not found")
     }

metal/paddle-mobile/paddle-mobile/Common/PaddleMobileUnitTest.swift → metal/paddle-mobile/paddle-mobile/Src/Common/PaddleMobileUnitTest.swift

@@ -324,9 +324,10 @@ public class PaddleMobileUnitTest {
         
         let param = ConvAddBatchNormReluTestParam.init(inInputTexture: inputeTexture, inOutputTexture: outputTexture, inMetalParam: metalParam, inFilterBuffer: filterBuffer, inBiaseBuffer: biaseBuffer, inNewScaleBuffer: newScalueBuffer, inNewBiaseBuffer: newBiaseBuffer, inFilterSize: filterSize)
         
+      let initContext = InitContext.init()
+      initContext.metalLoadMode = .LoadMetalInDefaultLib
         
-        
-        let convAddBnReluKernel = ConvAddBatchNormReluKernel<Float32>.init(device: device, testParam: param)
+      let convAddBnReluKernel = ConvAddBatchNormReluKernel<Float32>.init(device: device, testParam: param, initContext: initContext)
         
         convAddBnReluKernel.test(commandBuffer: buffer, param: param)
         

metal/paddle-mobile/paddle-mobile/Common/Types.swift → metal/paddle-mobile/paddle-mobile/Src/Common/Types.swift

@@ -252,11 +252,11 @@ extension InputTexture: Variant {
 extension MTLTexture where Self: Variant {
 }
 
-class FetchHolder: Variant {
+public class FetchHolder: Variant {
   var resultBuffer: MTLBuffer?
-  var dim: Dim
-  var capacity: Int
-  var paddedCapacity: Int
+  public var dim: Dim
+  public var capacity: Int
+  public var paddedCapacity: Int
   
   init(inPaddedCapacity: Int, inDim: Dim) {
     paddedCapacity = inPaddedCapacity
@@ -264,7 +264,7 @@ class FetchHolder: Variant {
     dim = inDim
   }
   
-  func initBuffer(device: MTLDevice) {
+  public func initBuffer(device: MTLDevice) {
     resultBuffer = device.makeBuffer(length: paddedCapacity * 4, options: [])
   }
   
@@ -278,12 +278,12 @@ class FetchHolder: Variant {
 }
 
 extension FetchHolder: CustomStringConvertible, CustomDebugStringConvertible {
-  var description: String {
+  public var description: String {
     fatalError()
 //    return "\(result)"
   }
   
-  var debugDescription: String {
+  public var debugDescription: String {
     fatalError()
 //    return "\(result)"
   }

metal/paddle-mobile/paddle-mobile/Operators/Base/OpCreator.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/Base/OpCreator.swift

@@ -27,19 +27,19 @@ class OpCreator<P: PrecisionType> {
         }
     }
     
-    func creat(device: MTLDevice, opDesc: OpDesc, scope: Scope) throws -> Runable & InferShaperable {
+  func creat(device: MTLDevice, opDesc: OpDesc, scope: Scope, initContext: InitContext) throws -> Runable & InferShaperable {
         guard let opCreator = opCreators[opDesc.type] else {
             throw PaddleMobileError.opError(message: "there is no " + opDesc.type + " yet")
         }
         
         do {
-            return try opCreator(device, opDesc, scope)
+            return try opCreator(device, opDesc, scope, initContext)
         } catch let error {
             throw error
         }
     }
     
-    let opCreators: [String : (MTLDevice, OpDesc, Scope) throws -> Runable & InferShaperable] =
+    let opCreators: [String : (MTLDevice, OpDesc, Scope, InitContext) throws -> Runable & InferShaperable] =
         [gConvType                  :     ConvOp<P>.creat,
          gBatchNormType             :     BatchNormOp<P>.creat,
          gReluType                  :     ReluOp<P>.creat,

metal/paddle-mobile/paddle-mobile/Operators/Base/Operator.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/Base/Operator.swift

@@ -31,7 +31,7 @@ protocol Runable {
   func run(device: MTLDevice, buffer: MTLCommandBuffer) throws
   func runImpl(device: MTLDevice,buffer: MTLCommandBuffer) throws
   func delogOutput()
-  func inputVariant() -> [String : [Variant]]
+  func inputVariant() -> [String : [MTLBuffer]]
   func computeMiddleResult(device: MTLDevice, buffer: MTLCommandBuffer)
 }
 
@@ -44,7 +44,7 @@ extension Runable where Self: OperatorProtocol{
     }
   }
   
-  func inputVariant() -> [String : [Variant]] {
+  func inputVariant() -> [String : [MTLBuffer]] {
 //    return [:]
     fatalError(" op \(type) need implement inputVariant")
   }
@@ -59,15 +59,26 @@ extension Runable where Self: OperatorProtocol{
   }
 }
 
+public class InitContext {
+  /// metal 代码加载方式
+  var metalLoadMode: MetalLoadMode = .LoadMetalInDefaultLib
+  /// 当 metalLoadMode 为 LoadMetalInCustomMetalLib 时， metal library 路径不能为空
+  var metalLibPath: String? = nil
+  init() {
+    metalLoadMode = .LoadMetalInDefaultLib
+    metalLibPath = nil
+  }
+}
+
 protocol Creator where Self: OperatorProtocol{
   associatedtype OpType: OperatorProtocol & Runable & InferShaperable
-  static func creat(device: MTLDevice, opDesc: OpDesc, inScope: Scope) throws -> OpType
+  static func creat(device: MTLDevice, opDesc: OpDesc, inScope: Scope, initContext: InitContext) throws -> OpType
 }
 
 extension Creator where Self: OperatorProtocol {
-  static func creat(device: MTLDevice, opDesc: OpDesc, inScope: Scope) throws -> OpType {
+  static func creat(device: MTLDevice, opDesc: OpDesc, inScope: Scope, initContext: InitContext) throws -> OpType {
     do {
-      return try OpType.provide(device:device, opDesc: opDesc, inScope: inScope)
+      return try OpType.provide(device:device, opDesc: opDesc, inScope: inScope, initContext: initContext)
     } catch let error {
       throw error
     }
@@ -89,13 +100,13 @@ protocol OperatorProtocol {
   var attrs: [String : Attr] { get }
   var para: ParamType { get }
   var kernel: KerType { get }
-  init(device: MTLDevice, opDesc: OpDesc, inScope: Scope) throws
+  init(device: MTLDevice, opDesc: OpDesc, inScope: Scope, initContext: InitContext) throws
 }
 
 extension OperatorProtocol {
-  static func provide(device: MTLDevice, opDesc: OpDesc, inScope: Scope) throws -> Self {
+  static func provide(device: MTLDevice, opDesc: OpDesc, inScope: Scope, initContext: InitContext) throws -> Self {
     do {
-      return try Self.init(device: device, opDesc: opDesc, inScope: inScope)
+      return try Self.init(device: device, opDesc: opDesc, inScope: inScope, initContext: initContext)
     } catch let error {
       throw error
     }
@@ -103,18 +114,7 @@ extension OperatorProtocol {
 }
 
 class Operator <KernelType:  Computable , ParameterType>: OperatorProtocol where KernelType.ParamType == ParameterType {
-  typealias ParamType = ParameterType
-  typealias KerType = KernelType
-  let type: String
-  let inputs: [String : [String]]
-  var paraInputs: [String : [String]]
-  let outpus: [String : [String]]
-  let attrs: [String : Attr]
-  let para: ParamType
-  let scope: Scope
-  var kernel: KerType
-  required init(device: MTLDevice, opDesc: OpDesc, inScope: Scope) throws {
-//    print("create op: \(opDesc.type)")
+  required init(device: MTLDevice, opDesc: OpDesc, inScope: Scope, initContext: InitContext) throws {
     type = opDesc.type
     scope = inScope
     inputs = opDesc.inputs
@@ -126,8 +126,19 @@ class Operator <KernelType:  Computable , ParameterType>: OperatorProtocol where
     } catch let error {
       throw error
     }
-    kernel = KernelType.init(device: device, param: para)
+    kernel = KernelType.init(device: device, param: para, initContext: initContext)
   }
+  
+  typealias ParamType = ParameterType
+  typealias KerType = KernelType
+  let type: String
+  let inputs: [String : [String]]
+  var paraInputs: [String : [String]]
+  let outpus: [String : [String]]
+  let attrs: [String : Attr]
+  let para: ParamType
+  let scope: Scope
+  var kernel: KerType
 }
 
 // op infos

metal/paddle-mobile/paddle-mobile/Operators/ConvAddOp.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/ConvAddOp.swift

@@ -111,6 +111,7 @@ class ConvAddOp<P: PrecisionType>: Operator<ConvAddKernel<P>, ConvAddParam<P>>,
 //    print(biase)
     
     print(" \(type) output: ")
+    print(para.output.metalTexture)
     print(para.output.metalTexture.toTensor(dim: (n: para.output.tensorDim[0], c: para.output.tensorDim[1], h: para.output.tensorDim[2], w: para.output.tensorDim[3])).strideArray())
   }
 }

metal/paddle-mobile/paddle-mobile/Operators/DepthwiseConvOp.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/DepthwiseConvOp.swift

@@ -17,14 +17,6 @@ import Foundation
 class DepthConvOp<P: PrecisionType>: Operator<ConvKernel<P>, ConvParam<P>>, Runable, Creator, InferShaperable {
 
   typealias OpType = DepthConvOp<P>
-
-  required init(device: MTLDevice, opDesc: OpDesc, inScope: Scope) throws {
-    do {
-      try super.init(device: device, opDesc: opDesc, inScope: inScope)
-    } catch let error {
-      throw error
-    }
-  }
   
   func inferShape() {
     let inDims = para.input.dim

metal/paddle-mobile/paddle-mobile/Operators/ElementwiseAddOp.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/ElementwiseAddOp.swift

@@ -32,7 +32,7 @@ class ElementwiseAddParam<P: PrecisionType>: OpParam {
       let device = inputX.metalTexture!.device
       inputY = Texture.init(device: device, inDim: tensorY.dim)
       let value: [P] = Array(UnsafeBufferPointer(start: tensorY.data.pointer, count: tensorY.dim.numel()))
-      inputY.metalTexture = device.tensor2texture(value: value, dim: tensorY.dim.dims, transpose: [0, 1, 2, 3], inComputePrecision: computePrecision)
+      inputY.metalTexture = device.tensor2texture(value: value, dim: tensorY.dim.dims, transpose: [0, 1, 2, 3], inComputePrecision: GlobalConfig.shared.computePrecision)
     }
     
 //    required init(device: MTLDevice, param: ElementwiseAddParam<P>) {

metal/paddle-mobile/paddle-mobile/Operators/ElementwiseAddPreluOp.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/ElementwiseAddPreluOp.swift

@@ -34,7 +34,7 @@ class ElementwiseAddPreluParam<P: PrecisionType>: OpParam {
       let device = inputX.metalTexture!.device
       inputY = Texture.init(device: device, inDim: tensorY.dim)
       let value: [P] = Array(UnsafeBufferPointer(start: tensorY.data.pointer, count: tensorY.dim.numel()))
-      inputY.metalTexture = device.tensor2texture(value: value, dim: tensorY.dim.dims, transpose: [0, 1, 2, 3], inComputePrecision: computePrecision)
+      inputY.metalTexture = device.tensor2texture(value: value, dim: tensorY.dim.dims, transpose: [0, 1, 2, 3], inComputePrecision: GlobalConfig.shared.computePrecision)
     }
     
     //    required init(device: MTLDevice, param: ElementwiseAddParam<P>) {

metal/paddle-mobile/paddle-mobile/Operators/FeedOp.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/FeedOp.swift

@@ -63,7 +63,7 @@ class FeedOp<P: PrecisionType>: Operator<Texture2DTo2DArrayKernel<P>, FeedParam<
   
   func delogOutput() {
     print(" \(type) output: ")
-    
+    print(para.output.metalTexture)
     print(para.output.metalTexture.toTensor(dim: (n: para.output.padToFourDim[0], c: para.output.padToFourDim[3], h: para.output.padToFourDim[2], w: para.output.padToFourDim[1])).strideArray())
   }
 }

metal/paddle-mobile/paddle-mobile/Operators/FetchOp.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/FetchOp.swift

@@ -45,17 +45,17 @@ class FetchKernel<P: PrecisionType>: Kernel, Computable {
     encoder.endEncoding()
   }
   
-  required init(device: MTLDevice, param: FetchParam<P>) {
+  required init(device: MTLDevice, param: FetchParam<P>, initContext: InitContext) {
     param.output.initBuffer(device: device)
-    if computePrecision == .Float16 {
+    if GlobalConfig.shared.computePrecision == .Float16 {
       if param.input.transpose == [0, 2, 3, 1] {
-        super.init(device: device, inFunctionName: "fetch_half")
+        super.init(device: device, inFunctionName: "fetch_half", initContext: initContext)
       } else {
         fatalError(" not support ")
       }
-    } else if computePrecision == .Float32 {
+    } else if GlobalConfig.shared.computePrecision == .Float32 {
       if param.input.transpose == [0, 2, 3, 1] {
-        super.init(device: device, inFunctionName: "fetch_float")
+        super.init(device: device, inFunctionName: "fetch_float", initContext: initContext)
       } else {
         fatalError(" not support ")
       }

metal/paddle-mobile/paddle-mobile/Operators/Kernels/Base/Kernel.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/Kernels/Base/Kernel.swift

@@ -21,14 +21,14 @@ public protocol TestParam {
 public protocol Testable {
   associatedtype TestParamType: TestParam
   func test(commandBuffer: MTLCommandBuffer, param: TestParamType)
-  init(device: MTLDevice, testParam: TestParamType)
+  init(device: MTLDevice, testParam: TestParamType, initContext: InitContext)
 }
 
 
 protocol Computable {
   associatedtype ParamType: OpParam
   func compute(commandBuffer: MTLCommandBuffer, param: ParamType) throws
-  init(device: MTLDevice, param: ParamType)
+  init(device: MTLDevice, param: ParamType, initContext: InitContext)
 }
 
 protocol KernelProtocol {
@@ -37,20 +37,20 @@ protocol KernelProtocol {
   
 }
 
-open class Kernel {
+@objc open class Kernel: NSObject{
   let pipline: MTLComputePipelineState
   let functionName: String
-  public init(device: MTLDevice, inFunctionName: String, usePaddleMobileLib: Bool = true) {
-    pipline = device.pipeLine(funcName: inFunctionName, inPaddleMobileLib: usePaddleMobileLib)
+  public init(device: MTLDevice, inFunctionName: String, usePaddleMobileLib: Bool = false, initContext: InitContext) {
+    pipline = device.pipeLine(funcName: inFunctionName, metalLoadMode: initContext.metalLoadMode, metalLibPath: initContext.metalLibPath)
     functionName = inFunctionName
   }
 }
 
-public struct Shape {
+@objc public class Shape: NSObject {
   public let width: Int
   public let height: Int
   public let channel: Int
-  public init(inWidth: Int, inHeight: Int, inChannel: Int){
+  @objc public init(inWidth: Int, inHeight: Int, inChannel: Int){
     width = inWidth
     height = inHeight
     channel = inChannel
@@ -60,16 +60,16 @@ public struct Shape {
 open class BufferToTextureKernel: Kernel {
   public let outputTexture: MTLTexture
   
-  public init(device: MTLDevice, outputDim: Shape, usePaddleMobileLib: Bool = false) {
+  public init(device: MTLDevice, outputDim: Shape, metalLoadMode: MetalLoadMode, metalLibPath: String?) {
     let textureDesc = MTLTextureDescriptor.init()
     textureDesc.textureType = .type2D
     textureDesc.width = outputDim.width
     textureDesc.height = outputDim.height
     textureDesc.depth = (outputDim.channel + 3) / 4
     
-    if computePrecision == .Float16 {
+    if GlobalConfig.shared.computePrecision == .Float16 {
       textureDesc.pixelFormat = .rgba16Float
-    } else if computePrecision == .Float32 {
+    } else if GlobalConfig.shared.computePrecision == .Float32 {
       textureDesc.pixelFormat = .rgba32Float
     } else {
       fatalError()
@@ -78,10 +78,13 @@ open class BufferToTextureKernel: Kernel {
     textureDesc.usage = [.shaderRead, .shaderWrite]
     textureDesc.storageMode = .shared
     outputTexture = device.makeTexture(descriptor: textureDesc) ?! " make texture error "
-    if computePrecision == .Float32 {
-      super.init(device: device, inFunctionName: "buffer_to_texture_kernel", usePaddleMobileLib: usePaddleMobileLib)
+    let initContext = InitContext.init()
+    initContext.metalLibPath = metalLibPath
+    initContext.metalLoadMode = metalLoadMode
+    if GlobalConfig.shared.computePrecision == .Float32 {
+      super.init(device: device, inFunctionName: "buffer_to_texture_kernel", initContext: initContext)
     } else {
-      super.init(device: device, inFunctionName: "buffer_to_texture_kernel_half", usePaddleMobileLib: usePaddleMobileLib)
+      super.init(device: device, inFunctionName: "buffer_to_texture_kernel_half", initContext: initContext)
     }
   }
   
@@ -98,19 +101,19 @@ open class BufferToTextureKernel: Kernel {
 
 }
 
-open class CusomKernel: Kernel {
+@objc open class CusomKernel: Kernel {
 
   public let outputTexture: MTLTexture
-  public init(device: MTLDevice, inFunctionName: String, outputDim: Shape, usePaddleMobileLib: Bool = false) {
+  public init(device: MTLDevice, inFunctionName: String, outputDim: Shape, metalLoadModel: MetalLoadMode, metalLibPath: String?) {
     let textureDesc = MTLTextureDescriptor.init()
     textureDesc.textureType = .type2D
     textureDesc.width = outputDim.width
     textureDesc.height = outputDim.height
     textureDesc.depth = (outputDim.channel + 3) / 4
     
-    if computePrecision == .Float16 {
+    if GlobalConfig.shared.computePrecision == .Float16 {
       textureDesc.pixelFormat = .rgba16Float
-    } else if computePrecision == .Float32 {
+    } else if GlobalConfig.shared.computePrecision == .Float32 {
       textureDesc.pixelFormat = .rgba32Float
     } else {
       fatalError()
@@ -120,7 +123,10 @@ open class CusomKernel: Kernel {
     textureDesc.storageMode = .shared
     outputTexture = device.makeTexture(descriptor: textureDesc) ?! " make texture error "
     
-    super.init(device: device, inFunctionName: inFunctionName, usePaddleMobileLib: usePaddleMobileLib)
+    let context = InitContext.init()
+    context.metalLoadMode = metalLoadModel
+    context.metalLibPath = metalLibPath
+    super.init(device: device, inFunctionName: inFunctionName, initContext: context)
   }
   
   public func compute(inputTexuture: MTLTexture, commandBuffer: MTLCommandBuffer) throws {

metal/paddle-mobile/paddle-mobile/Operators/Kernels/BatchNormKernel.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/Kernels/BatchNormKernel.swift

@@ -15,7 +15,7 @@
 import Foundation
 
 class BatchNormKernel<P: PrecisionType>: Kernel, Computable {
-  required init(device: MTLDevice, param: BatchNormParam<P>) {
+  required init(device: MTLDevice, param: BatchNormParam<P>, initContext: InitContext) {
     let count = param.variance.dim.numel()
     let varianceP = param.variance.data.pointer
     let meanP = param.mean.data.pointer
@@ -27,13 +27,13 @@ class BatchNormKernel<P: PrecisionType>: Kernel, Computable {
       scaleP[i] = invStd * scaleP[i]
     }
 
-    param.bias.initBuffer(device: device, precision: computePrecision)
-    param.scale.initBuffer(device: device, precision: computePrecision)
-    param.output.initTexture(device: device, inTranspose: param.input.transpose, computePrecision: computePrecision)
-    if computePrecision == .Float32 {
-      super.init(device: device, inFunctionName: "batchnorm")
-    } else if computePrecision == .Float16 {
-      super.init(device: device, inFunctionName: "batchnorm_half")
+    param.bias.initBuffer(device: device, precision: GlobalConfig.shared.computePrecision)
+    param.scale.initBuffer(device: device, precision: GlobalConfig.shared.computePrecision)
+    param.output.initTexture(device: device, inTranspose: param.input.transpose, computePrecision: GlobalConfig.shared.computePrecision)
+    if GlobalConfig.shared.computePrecision == .Float32 {
+      super.init(device: device, inFunctionName: "batchnorm", initContext: initContext)
+    } else if GlobalConfig.shared.computePrecision == .Float16 {
+      super.init(device: device, inFunctionName: "batchnorm_half", initContext: initContext)
     } else {
       fatalError()
     }

metal/paddle-mobile/paddle-mobile/Operators/Kernels/BilinearInterpKernel.swift → metal/paddle-mobile/paddle-mobile/Src/Operators/Kernels/BilinearInterpKernel.swift

@@ -41,12 +41,12 @@ class BilinearInterpKernel<P: PrecisionType>: Kernel, Computable{
     encoder.endEncoding()
   }
   
-  required init(device: MTLDevice, param: BilinearInterpParam<P>) {
-    param.output.initTexture(device: device, inTranspose: param.input.transpose, computePrecision: computePrecision)
-    if computePrecision == .Float32 {
-      super.init(device: device, inFunctionName: "bilinear_interp_float")
-    } else if computePrecision == .Float16 {
-      super.init(device: device, inFunctionName: "bilinear_interp_half")
+  required init(device: MTLDevice, param: BilinearInterpParam<P>, initContext: InitContext) {
+    param.output.initTexture(device: device, inTranspose: param.input.transpose, computePrecision: GlobalConfig.shared.computePrecision)
+    if GlobalConfig.shared.computePrecision == .Float32 {
+      super.init(device: device, inFunctionName: "bilinear_interp_float", initContext: initContext)
+    } else if GlobalConfig.shared.computePrecision == .Float16 {
+      super.init(device: device, inFunctionName: "bilinear_interp_half", initContext: initContext)
     } else {
       fatalError()
     }