From e780b1dbf50c1ba66869c3d895f3f9b0757941d4 Mon Sep 17 00:00:00 2001 From: Your Name Date: Wed, 9 Oct 2024 11:04:36 -0700 Subject: [PATCH] - --- benchmarks/lightning/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmarks/lightning/main.py b/benchmarks/lightning/main.py index 7d218743..b5398478 100644 --- a/benchmarks/lightning/main.py +++ b/benchmarks/lightning/main.py @@ -1,5 +1,7 @@ #!/usr/bin/env python +from habana_frameworks.torch import hpu; hpu.init() + import argparse import os @@ -55,8 +57,6 @@ def prepare_voir(): return observer, bench_monitor def main(): - from habana_frameworks.torch import hpu; hpu.init() - rank = int(os.getenv("RANK", 0)) world_size = int(os.getenv("WORLD_SIZE", 1)) local_world_size = int(os.getenv("LOCAL_WORLD_SIZE", 1))