intel-ai-tce
diff --git a/‎.gitmodules
+3 b/‎.gitmodules
+3
diff --git a/‎README.md
+65-58 b/‎README.md
+65-58
diff --git a/‎benchmarks/README.md
+52-52 b/‎benchmarks/README.md
+52-52
diff --git a/‎benchmarks/common/base_benchmark_util.py
+21-6 b/‎benchmarks/common/base_benchmark_util.py
+21-6
diff --git a/‎benchmarks/common/platform_util.py
+1-1 b/‎benchmarks/common/platform_util.py
+1-1
@@ -0,0 +1,3 @@
+[submodule "models/aidd/pytorch/alphafold2/inference/alphafold"]
+	path = models/aidd/pytorch/alphafold2/inference/alphafold
+	url = https://github.com/deepmind/alphafold
@@ -1,7 +1,7 @@
 #
 # -*- coding: utf-8 -*-
 #
-# Copyright (c) 2018-2019 Intel Corporation
+# Copyright (c) 2018-2021 Intel Corporation
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -68,8 +68,8 @@ def _define_args(self):
 
         self._common_arg_parser.add_argument(
             "-p", "--precision",
-            help="Specify the model precision to use: fp32, int8, or bfloat16",
-            required=required_arg, choices=["fp32", "int8", "bfloat16"],
+            help="Specify the model precision to use: fp32, fp16, int8, or bfloat16",
+            required=required_arg, choices=["fp32", "fp16", "int8", "bfloat16"],
             dest="precision")
 
         self._common_arg_parser.add_argument(
@@ -172,7 +172,7 @@ def _define_args(self):
             dest="weight_sharing", action="store_true")
         self._common_arg_parser.add_argument(
             "--synthetic-data",
-            help="Enables synthetic data layer for some models where support exists",
+            help="Enables synthetic data layer for some models like SSD-ResNet34 where support exists",
             dest="synthetic_data", action="store_true")
 
         self._common_arg_parser.add_argument(
@@ -274,6 +274,12 @@ def _define_args(self):
             help="Additional command line arguments (prefix flag start with"
                  " '--').")
 
+        # Check if GPU is enabled.
+        self._common_arg_parser.add_argument(
+            "--gpu",
+            help="Run the benchmark script using GPU",
+            dest="gpu", action="store_true")
+
     def _validate_args(self):
         """validate the args and initializes platform_util"""
         # check if socket id is in socket number range
@@ -304,8 +310,9 @@ def _validate_args(self):
                              format(system_num_cores))
 
         if args.output_results and ((args.model_name != "resnet50" and
-                                    args.model_name != "resnet50v1_5") or args.precision != "fp32"):
-            raise ValueError("--output-results is currently only supported for resnet50 FP32 inference.")
+                                    args.model_name != "resnet50v1_5") or
+                                    (args.precision != "fp32" and args.precision != "fp16")):
+            raise ValueError("--output-results is currently only supported for resnet50 FP32 or FP16 inference.")
         elif args.output_results and (args.mode != "inference" or not args.data_location):
             raise ValueError("--output-results can only be used when running inference with a dataset.")
 
@@ -348,6 +355,14 @@ def _validate_args(self):
                       "This is less than the number of cores per socket on the system ({})".
                       format(args.socket_id, cpuset_len_for_socket, self._platform_util.num_cores_per_socket))
 
+        if args.gpu:
+            if args.socket_id != -1:
+                raise ValueError("--socket-id cannot be used with --gpu parameter.")
+            if args.num_intra_threads is not None:
+                raise ValueError("--num-intra-threads cannot be used with --gpu parameter.")
+            if args.num_inter_threads is not None:
+                raise ValueError("--num-inter-threads cannot be used with --gpu parameter.")
+
     def initialize_model(self, args, unknown_args):
         """Create model initializer for the specified model"""
         model_initializer = None
 
@@ -229,7 +229,7 @@ def _get_list_from_string_ranges(self, str_ranges):
                 start, end = section.split("-")
                 section_list = range(int(start), int(end) + 1)
                 result_list += section_list
-            elif(len(section)):
+            elif len(section):
                 # This section is either empty or just a single number and not a range
                 result_list.append(int(section))
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+[submodule "models/aidd/pytorch/alphafold2/inference/alphafold"]`
	`2`	`+ path = models/aidd/pytorch/alphafold2/inference/alphafold`
	`3`	`+ url = https://github.com/deepmind/alphafold`