lint

nod-ai · Feb 6, 2024 · acad468 · acad468
1 parent 42821b3
commit acad468
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core/shark_turbine/dynamo/passes.py b/core/shark_turbine/dynamo/passes.py
@@ -54,7 +54,9 @@
 
 # These decompositions either didnt exist or weren't required for 2.1.0
 if torch.__version__ > "2.1.0":
-    DEFAULT_DECOMPOSITIONS.append(torch.ops.aten._scaled_dot_product_flash_attention_for_cpu)
+    DEFAULT_DECOMPOSITIONS.append(
+        torch.ops.aten._scaled_dot_product_flash_attention_for_cpu
+    )
     DEFAULT_DECOMPOSITIONS.append(torch.ops.aten.unbind_int)