Browse Source

Merge pull request #778 from exo-explore/grpcio1.71.0

upgrade grpcio and grpcio-tools to 1.71.0
Alex Cheema 3 months ago
parent
commit
04d5dca18f
2 changed files with 3 additions and 2 deletions
  1. 1 0
      exo/topology/device_capabilities.py
  2. 2 2
      setup.py

+ 1 - 0
exo/topology/device_capabilities.py

@@ -56,6 +56,7 @@ CHIP_FLOPS = {
   "Apple M3": DeviceFlops(fp32=3.55*TFLOPS, fp16=7.10*TFLOPS, int8=14.20*TFLOPS),
   "Apple M3 Pro": DeviceFlops(fp32=4.97*TFLOPS, fp16=9.94*TFLOPS, int8=19.88*TFLOPS),
   "Apple M3 Max": DeviceFlops(fp32=14.20*TFLOPS, fp16=28.40*TFLOPS, int8=56.80*TFLOPS),
+  "Apple M3 Ultra": DeviceFlops(fp32=54.26*TFLOPS, fp16=108.52*TFLOPS, int8=217.04*TFLOPS),
   "Apple M4": DeviceFlops(fp32=4.26*TFLOPS, fp16=8.52*TFLOPS, int8=17.04*TFLOPS),
   "Apple M4 Pro": DeviceFlops(fp32=5.72*TFLOPS, fp16=11.44*TFLOPS, int8=22.88*TFLOPS),
   "Apple M4 Max": DeviceFlops(fp32=18.03*TFLOPS, fp16=36.07*TFLOPS, int8=72.14*TFLOPS),

+ 2 - 2
setup.py

@@ -9,8 +9,8 @@ install_requires = [
   "aiohttp==3.10.11",
   "aiohttp_cors==0.7.0",
   "aiofiles==24.1.0",
-  "grpcio==1.67.0",
-  "grpcio-tools==1.67.0",
+  "grpcio==1.71.0",
+  "grpcio-tools==1.71.0",
   "Jinja2==3.1.4",
   "numpy==2.0.0",
   "nuitka==2.5.1",