re-arranged code to be more specific

FrostyTheSouthernSnowman · Aug 29, 2023 · f5bd049 · f5bd049
1 parent d0b9f68
commit f5bd049
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/tinygrad/runtime/ops_metal.py b/tinygrad/runtime/ops_metal.py
@@ -1,16 +1,17 @@
-# pip3 install pyobjc-framework-Metal pyobjc-framework-Cocoa pyobjc-framework-libdispatch
+import os, subprocess, pathlib, functools, ctypes
+
 try:
-    import os, subprocess, pathlib, functools, ctypes
     import Metal, Cocoa, libdispatch # type: ignore
-    from typing import List, Any
-    from tinygrad.codegen.linearizer import LinearizerOptions
-    from tinygrad.renderer.cstyle import uops_to_cstyle, CStyleLanguage
-    from tinygrad.helpers import prod, getenv, DEBUG, DType, dtypes
-    from tinygrad.ops import Compiled
-    from tinygrad.runtime.lib import RawBufferMapped, LRUAllocator
 except ImportError as e:
     raise ImportError(f"{e.name}. \n If you may want to try:\n\tpip3 install pyobjc-framework-Metal pyobjc-framework-Cocoa pyobjc-framework-libdispatch\n")
 
+from typing import List, Any
+from tinygrad.codegen.linearizer import LinearizerOptions
+from tinygrad.renderer.cstyle import uops_to_cstyle, CStyleLanguage
+from tinygrad.helpers import prod, getenv, DEBUG, DType, dtypes
+from tinygrad.ops import Compiled
+from tinygrad.runtime.lib import RawBufferMapped, LRUAllocator
+
 METAL_XCODE = getenv("METAL_XCODE")
 
 class MetalAllocator(LRUAllocator):