File tree 2 files changed +3
-1
lines changed
2 files changed +3
-1
lines changed Original file line number Diff line number Diff line change 1
- 7d7c14e898eca3fe66138d2a9445755a9270b800
1
+ 2e032c6b0de960dee554dcb08126ace718b14c6d
Original file line number Diff line number Diff line change @@ -932,13 +932,15 @@ def quantized_model(self) -> nn.Module:
932
932
libs = glob .glob (f"{ torchao_build_path } /cmake-out/lib/libtorchao_ops_aten.*" )
933
933
libs = list (filter (lambda l : (l .endswith ("so" ) or l .endswith ("dylib" )), libs ))
934
934
torch .ops .load_library (libs [0 ])
935
+ print ("Loaded torchao cpu ops." )
935
936
except Exception as e :
936
937
print ("Unabled to load torchao cpu ops library. Slow fallback kernels will be used." )
937
938
938
939
try :
939
940
libname = "libtorchao_ops_mps_aten.dylib"
940
941
libpath = f"{ torchao_build_path } /cmake-out/lib/{ libname } "
941
942
torch .ops .load_library (libpath )
943
+ print ("Loaded torchao mps ops." )
942
944
except Exception as e :
943
945
print ("Unabled to load torchao mps ops library." )
944
946
You can’t perform that action at this time.
0 commit comments