up

Author: metascroy

torchao/experimental/ops/mps/test/test_lowbit.py

@@ -79,30 +79,30 @@
 
 if __name__ == "__main__":
     print("RUNNING UNIT TESTS")
-    try:
-        print("TRYING")
-        for nbit in range(1, 8):
-            print("NBIT", nbit)
-            getattr(torch.ops.torchao, f"_linear_fp_act_{nbit}bit_weight")
-            getattr(torch.ops.torchao, f"_pack_weight_{nbit}bit")
-    except AttributeError:
-        try:
-            print("LOADING LIB")
-            libname = "libtorchao_ops_mps_aten.dylib"
-            libpath = os.path.abspath(os.path.join(os.path.dirname(__file__), "../cmake-out/lib/", libname))
-            print("AT ", libpath)
-            torch.ops.load_library(libpath)
-            print("LOADED")
-        except Exception as e:
-            print("FAILED TO LOAD")
-            raise e
-            # raise RuntimeError(f"Failed to load library {libpath}")
-        else:
-            try:
-                for nbit in range(1, 8):
-                    getattr(torch.ops.torchao, f"_linear_fp_act_{nbit}bit_weight")
-                    getattr(torch.ops.torchao, f"_pack_weight_{nbit}bit")
-            except AttributeError as e:
-                raise e
+    # try:
+    #     print("TRYING")
+    #     for nbit in range(1, 8):
+    #         print("NBIT", nbit)
+    #         getattr(torch.ops.torchao, f"_linear_fp_act_{nbit}bit_weight")
+    #         getattr(torch.ops.torchao, f"_pack_weight_{nbit}bit")
+    # except AttributeError:
+    #     try:
+    #         print("LOADING LIB")
+    #         libname = "libtorchao_ops_mps_aten.dylib"
+    #         libpath = os.path.abspath(os.path.join(os.path.dirname(__file__), "../cmake-out/lib/", libname))
+    #         print("AT ", libpath)
+    #         torch.ops.load_library(libpath)
+    #         print("LOADED")
+    #     except Exception as e:
+    #         print("FAILED TO LOAD")
+    #         raise e
+    #         # raise RuntimeError(f"Failed to load library {libpath}")
+    #     else:
+    #         try:
+    #             for nbit in range(1, 8):
+    #                 getattr(torch.ops.torchao, f"_linear_fp_act_{nbit}bit_weight")
+    #                 getattr(torch.ops.torchao, f"_pack_weight_{nbit}bit")
+    #         except AttributeError as e:
+    #             raise e
 
     # unittest.main()