update torchao pin: optimized shaders (#1428)

Author: manuelcandales

install/.pins/torchao-pin.txt

@@ -1 +1 @@
-7d7c14e898eca3fe66138d2a9445755a9270b800
+2e032c6b0de960dee554dcb08126ace718b14c6d

torchchat/utils/quantize.py

@@ -832,6 +832,7 @@ def quantized_model(self) -> nn.Module:
         libs = glob.glob(f"{torchao_build_path}/cmake-out/lib/libtorchao_ops_aten.*")
         libs = list(filter(lambda l: (l.endswith("so") or l.endswith("dylib")), libs))
         torch.ops.load_library(libs[0])
+        print("Loaded torchao cpu ops.")
     except Exception as e:
         print(
             "Unabled to load torchao cpu ops library. Slow fallback kernels will be used."
@@ -841,6 +842,7 @@ def quantized_model(self) -> nn.Module:
         libname = "libtorchao_ops_mps_aten.dylib"
         libpath = f"{torchao_build_path}/cmake-out/lib/{libname}"
         torch.ops.load_library(libpath)
+        print("Loaded torchao mps ops.")
     except Exception as e:
         print("Unabled to load torchao mps ops library.")