black lint formating

nod-ai · Jun 21, 2023 · 6fa8404 · 6fa8404
1 parent 89d3c17
commit 6fa8404
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/shark/iree_utils/compile_utils.py b/shark/iree_utils/compile_utils.py
@@ -51,7 +51,7 @@ def get_iree_device_args(device, extra_args=[]):
         )
     if device_uri[0] == "metal":
         from shark.iree_utils.metal_utils import get_iree_metal_args
-        
+
         return get_iree_metal_args(
             device_num=device_num, extra_args=extra_args
         )

diff --git a/shark/iree_utils/metal_utils.py b/shark/iree_utils/metal_utils.py
@@ -91,6 +91,7 @@ def get_metal_triple_flag(device_name="", device_num=0, extra_args=[]):
     print(f"Target : {metal_device}")
     return None
 
+
 def get_iree_metal_args(device_num=0, extra_args=[]):
     # res_metal_flag = ["--iree-flow-demote-i64-to-i32"]
 
@@ -101,7 +102,7 @@ def get_iree_metal_args(device_num=0, extra_args=[]):
             print(f"Using target triple {arg} from command line args")
             meatal_triple_flag = arg
             break
-    
+
     if metal_triple_flag is None:
         meatal_triple_flag = get_metal_triple_flag(
             device_num=device_num, extra_args=extra_args