@@ -4655,6 +4655,8 @@ def test_llama3_2_1b(self):
46554655 self .build_folder ,
46564656 "--model" ,
46574657 self .model ,
4658+ "--target" ,
4659+ self .target ,
46584660 "--checkpoint" ,
46594661 f"{ self .llama_artifacts } /consolidated.00.pth" ,
46604662 "--params" ,
@@ -4805,6 +4807,8 @@ def test_llama_stories_110m(self):
48054807 self .build_folder ,
48064808 "--model" ,
48074809 self .model ,
4810+ "--target" ,
4811+ self .target ,
48084812 "--checkpoint" ,
48094813 f"{ self .llama_artifacts } /stories110M.pt" ,
48104814 "--params" ,
@@ -5174,6 +5178,8 @@ def test_albert(self):
51745178 self .device ,
51755179 "--model" ,
51765180 self .model ,
5181+ "--target" ,
5182+ self .target ,
51775183 "--ip" ,
51785184 self .ip ,
51795185 "--port" ,
@@ -5210,6 +5216,8 @@ def test_bert(self):
52105216 self .device ,
52115217 "--model" ,
52125218 self .model ,
5219+ "--target" ,
5220+ self .target ,
52135221 "--ip" ,
52145222 self .ip ,
52155223 "--port" ,
@@ -5247,6 +5255,8 @@ def test_conv_former(self):
52475255 self .device ,
52485256 "--model" ,
52495257 self .model ,
5258+ "--target" ,
5259+ self .target ,
52505260 "--ip" ,
52515261 self .ip ,
52525262 "--port" ,
@@ -5285,6 +5295,8 @@ def test_cvt(self):
52855295 self .device ,
52865296 "--model" ,
52875297 self .model ,
5298+ "--target" ,
5299+ self .target ,
52885300 "--ip" ,
52895301 self .ip ,
52905302 "--port" ,
@@ -5322,6 +5334,8 @@ def test_deit(self):
53225334 self .device ,
53235335 "--model" ,
53245336 self .model ,
5337+ "--target" ,
5338+ self .target ,
53255339 "--ip" ,
53265340 self .ip ,
53275341 "--port" ,
@@ -5359,6 +5373,8 @@ def test_dino_v2(self):
53595373 self .device ,
53605374 "--model" ,
53615375 self .model ,
5376+ "--target" ,
5377+ self .target ,
53625378 "--ip" ,
53635379 self .ip ,
53645380 "--port" ,
@@ -5396,6 +5412,8 @@ def test_distilbert(self):
53965412 self .device ,
53975413 "--model" ,
53985414 self .model ,
5415+ "--target" ,
5416+ self .target ,
53995417 "--ip" ,
54005418 self .ip ,
54015419 "--port" ,
@@ -5431,6 +5449,8 @@ def test_dit(self):
54315449 self .device ,
54325450 "--model" ,
54335451 self .model ,
5452+ "--target" ,
5453+ self .target ,
54345454 "--ip" ,
54355455 self .ip ,
54365456 "--port" ,
@@ -5468,6 +5488,8 @@ def test_efficientnet(self):
54685488 self .device ,
54695489 "--model" ,
54705490 self .model ,
5491+ "--target" ,
5492+ self .target ,
54715493 "--ip" ,
54725494 self .ip ,
54735495 "--port" ,
@@ -5507,6 +5529,8 @@ def test_efficientSAM(self):
55075529 self .device ,
55085530 "--model" ,
55095531 self .model ,
5532+ "--target" ,
5533+ self .target ,
55105534 "--oss_repo" ,
55115535 self .oss_repo ,
55125536 "--pretrained_weight" ,
@@ -5547,6 +5571,8 @@ def test_esrgan(self):
55475571 self .device ,
55485572 "--model" ,
55495573 self .model ,
5574+ "--target" ,
5575+ self .target ,
55505576 "--default_dataset" ,
55515577 "--oss_repo" ,
55525578 self .oss_repo ,
@@ -5587,6 +5613,8 @@ def test_eurobert(self):
55875613 self .device ,
55885614 "--model" ,
55895615 self .model ,
5616+ "--target" ,
5617+ self .target ,
55905618 "--ip" ,
55915619 self .ip ,
55925620 "--port" ,
@@ -5625,6 +5653,8 @@ def test_fastvit(self):
56255653 self .device ,
56265654 "--model" ,
56275655 self .model ,
5656+ "--target" ,
5657+ self .target ,
56285658 "--oss_repo" ,
56295659 self .oss_repo ,
56305660 "--pretrained_weight" ,
@@ -5667,6 +5697,8 @@ def test_fbnet(self):
56675697 self .device ,
56685698 "--model" ,
56695699 self .model ,
5700+ "--target" ,
5701+ self .target ,
56705702 "--ip" ,
56715703 self .ip ,
56725704 "--port" ,
@@ -5705,6 +5737,8 @@ def test_focalnet(self):
57055737 self .device ,
57065738 "--model" ,
57075739 self .model ,
5740+ "--target" ,
5741+ self .target ,
57085742 "--ip" ,
57095743 self .ip ,
57105744 "--port" ,
@@ -5745,6 +5779,8 @@ def test_gMLP(self):
57455779 self .device ,
57465780 "--model" ,
57475781 self .model ,
5782+ "--target" ,
5783+ self .target ,
57485784 "--ip" ,
57495785 self .ip ,
57505786 "--port" ,
@@ -5784,6 +5820,8 @@ def test_mobilevit_v2(self):
57845820 self .device ,
57855821 "--model" ,
57865822 self .model ,
5823+ "--target" ,
5824+ self .target ,
57875825 "--ip" ,
57885826 self .ip ,
57895827 "--port" ,
@@ -5824,6 +5862,8 @@ def test_mobilevit_v1(self):
58245862 self .device ,
58255863 "--model" ,
58265864 self .model ,
5865+ "--target" ,
5866+ self .target ,
58275867 "--ip" ,
58285868 self .ip ,
58295869 "--port" ,
@@ -5901,6 +5941,8 @@ def test_regnet(self):
59015941 self .device ,
59025942 "--model" ,
59035943 self .model ,
5944+ "--target" ,
5945+ self .target ,
59045946 "--ip" ,
59055947 self .ip ,
59065948 "--port" ,
@@ -5940,6 +5982,8 @@ def test_retinanet(self):
59405982 self .device ,
59415983 "--model" ,
59425984 self .model ,
5985+ "--target" ,
5986+ self .target ,
59435987 "--dataset" ,
59445988 self .image_dataset ,
59455989 "--ip" ,
@@ -5978,6 +6022,8 @@ def test_roberta(self):
59786022 self .device ,
59796023 "--model" ,
59806024 self .model ,
6025+ "--target" ,
6026+ self .target ,
59816027 "--ip" ,
59826028 self .ip ,
59836029 "--port" ,
@@ -6015,6 +6061,8 @@ def test_squeezenet(self):
60156061 self .device ,
60166062 "--model" ,
60176063 self .model ,
6064+ "--target" ,
6065+ self .target ,
60186066 "--ip" ,
60196067 self .ip ,
60206068 "--port" ,
@@ -6051,6 +6099,8 @@ def test_ssd300_vgg16(self):
60516099 self .device ,
60526100 "--model" ,
60536101 self .model ,
6102+ "--target" ,
6103+ self .target ,
60546104 "--oss_repo" ,
60556105 self .oss_repo ,
60566106 "--pretrained_weight" ,
@@ -6091,6 +6141,8 @@ def test_swin_transformer(self):
60916141 self .device ,
60926142 "--model" ,
60936143 self .model ,
6144+ "--target" ,
6145+ self .target ,
60946146 "--ip" ,
60956147 self .ip ,
60966148 "--port" ,
@@ -6128,6 +6180,8 @@ def test_t5(self):
61286180 self .device ,
61296181 "--model" ,
61306182 self .model ,
6183+ "--target" ,
6184+ self .target ,
61316185 "--ip" ,
61326186 self .ip ,
61336187 "--port" ,
@@ -6163,6 +6217,8 @@ def test_whisper(self):
61636217 self .device ,
61646218 "--model" ,
61656219 self .model ,
6220+ "--target" ,
6221+ self .target ,
61666222 "--ip" ,
61676223 self .ip ,
61686224 "--port" ,
@@ -6430,6 +6486,8 @@ def test_mobilenet_v2(self):
64306486 self .device ,
64316487 "--model" ,
64326488 self .model ,
6489+ "--target" ,
6490+ self .target ,
64336491 "--ip" ,
64346492 self .ip ,
64356493 "--port" ,
@@ -6470,6 +6528,8 @@ def test_mobilenet_v3(self):
64706528 self .device ,
64716529 "--model" ,
64726530 self .model ,
6531+ "--target" ,
6532+ self .target ,
64736533 "--ip" ,
64746534 self .ip ,
64756535 "--port" ,
@@ -6510,6 +6570,8 @@ def test_inception_v3(self):
65106570 self .device ,
65116571 "--model" ,
65126572 self .model ,
6573+ "--target" ,
6574+ self .target ,
65136575 "--ip" ,
65146576 self .ip ,
65156577 "--port" ,
@@ -6550,6 +6612,8 @@ def test_inception_v4(self):
65506612 self .device ,
65516613 "--model" ,
65526614 self .model ,
6615+ "--target" ,
6616+ self .target ,
65536617 "--ip" ,
65546618 self .ip ,
65556619 "--port" ,
@@ -6590,6 +6654,8 @@ def test_vit(self):
65906654 self .device ,
65916655 "--model" ,
65926656 self .model ,
6657+ "--target" ,
6658+ self .target ,
65936659 "--ip" ,
65946660 self .ip ,
65956661 "--port" ,
@@ -6628,6 +6694,8 @@ def test_edsr(self):
66286694 self .device ,
66296695 "--model" ,
66306696 self .model ,
6697+ "--target" ,
6698+ self .target ,
66316699 "--default_dataset" ,
66326700 "--ip" ,
66336701 self .ip ,
@@ -6667,6 +6735,8 @@ def test_deeplab_v3(self):
66676735 self .device ,
66686736 "--model" ,
66696737 self .model ,
6738+ "--target" ,
6739+ self .target ,
66706740 "--download" ,
66716741 "--ip" ,
66726742 self .ip ,
@@ -6708,6 +6778,8 @@ def test_mobilebert(self):
67086778 self .device ,
67096779 "--model" ,
67106780 self .model ,
6781+ "--target" ,
6782+ self .target ,
67116783 "--pretrained_weight" ,
67126784 self .pretrained_weight ,
67136785 "--ip" ,
@@ -6749,6 +6821,8 @@ def test_ptq_mobilebert(self):
67496821 self .device ,
67506822 "--model" ,
67516823 self .model ,
6824+ "--target" ,
6825+ self .target ,
67526826 "--pretrained_weight" ,
67536827 self .pretrained_weight ,
67546828 "--ptq" ,
@@ -6791,6 +6865,8 @@ def test_wav2letter(self):
67916865 self .device ,
67926866 "--model" ,
67936867 self .model ,
6868+ "--target" ,
6869+ self .target ,
67946870 "--pretrained_weight" ,
67956871 self .pretrained_weight ,
67966872 "--ip" ,
@@ -6863,6 +6939,8 @@ def test_custom_op(self):
68636939 self .device ,
68646940 "--model" ,
68656941 self .model ,
6942+ "--target" ,
6943+ self .target ,
68666944 "--ip" ,
68676945 self .ip ,
68686946 "--port" ,
@@ -6895,6 +6973,8 @@ def test_debugger_generate_optrace(self):
68956973 self .device ,
68966974 "--model" ,
68976975 self .model ,
6976+ "--target" ,
6977+ self .target ,
68986978 "--ip" ,
68996979 self .ip ,
69006980 "--port" ,
@@ -6972,6 +7052,8 @@ def test_cli(self):
69727052 f"{ tmp_dir } /e_out" ,
69737053 "--model" ,
69747054 self .model ,
7055+ "--target" ,
7056+ self .target ,
69757057 "--device" ,
69767058 self .device ,
69777059 "--build_folder" ,
@@ -7051,19 +7133,6 @@ def setup_environment():
70517133 default = "" ,
70527134 type = str ,
70537135 )
7054- parser .add_argument (
7055- "-t" ,
7056- "--target" ,
7057- help = "Target platform for deployment" ,
7058- default = "aarch64-android" ,
7059- type = str ,
7060- )
7061- parser .add_argument (
7062- "--pre_gen_pte" ,
7063- help = "Run the pre-generated pte in the given directory." ,
7064- type = str ,
7065- )
7066-
70677136 parser .add_argument (
70687137 "--llama_artifacts" ,
70697138 help = "A folder that contains: weight, tokenizer, and params." ,
0 commit comments