Skip to content

Commit d91f463

Browse files
committed
Update reformating
1 parent aca1112 commit d91f463

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -559,13 +559,13 @@ def preprocess_gpu_flags():
559559
print("error: linux build requires --cudnn-home and --cuda-home")
560560

561561
if FLAGS.tensorrt_home is None:
562-
if target_platform() == "rhel":
563-
if platform.machine().lower() == "aarch64":
564-
FLAGS.tensorrt_home = "/usr/local/cuda/targets/sbsa-linux/"
565-
else:
566-
FLAGS.tensorrt_home = "/usr/local/cuda/targets/x86_64-linux/"
567-
else:
568-
FLAGS.tensorrt_home = "/usr/src/tensorrt"
562+
if target_platform() == "rhel":
563+
if platform.machine().lower() == "aarch64":
564+
FLAGS.tensorrt_home = "/usr/local/cuda/targets/sbsa-linux/"
565+
else:
566+
FLAGS.tensorrt_home = "/usr/local/cuda/targets/x86_64-linux/"
567+
else:
568+
FLAGS.tensorrt_home = "/usr/src/tensorrt"
569569

570570

571571
if __name__ == "__main__":
@@ -654,4 +654,4 @@ def preprocess_gpu_flags():
654654
FLAGS.ort_openvino = None
655655
dockerfile_for_windows(FLAGS.output)
656656
else:
657-
dockerfile_for_linux(FLAGS.output)
657+
dockerfile_for_linux(FLAGS.output)

0 commit comments

Comments
 (0)