Skip to content

Dual ANE Cluster on Ultra models #8

Description

@Anemll

We have received M3 Ultra and M1 Ultra reports for parallel model execution.
Unfortunately, neither shows bandwidth increase.
Either ANE service is not placing execution on different clusters, or memory I/O is somehow limited to 120GB/s for ANE on mac studio Ultra models.

We'll need to add some TOPS and FLOPS benchmarks to investigate further or get direct access to M3 Ultra units

{
"timestamp": "2025-03-12 09:33:07",
"system_info": {
"os": {
"name": "Darwin",
"version": "Darwin Kernel Version 24.3.0: Thu Jan 2 20:31:43 PST 2025; root:xnu-11215.81.4~4/RELEASE_ARM64_T6031",
"release": "24.3.0"
},
"cpu": {
"brand": "Apple M3 Ultra",
"architecture": "arm64",
"cores": 32,
"threads": 32
},
"ram": {
"total_gb": 512.0,
"available_gb": 487.37
},
"python_version": "3.12.9",
"mac_model": "Mac15,14",
"macos_version": "macOS Sequoia 15.3 (24D2059)",
"apple_silicon": true,
"neural_engine": {
"available": true,
"cores": "Unknown",
"tflops": "Unknown"
}
},
"results": [
{
"model_name": "llama_lm_head",
"backend": "ANE",
"inference_time_ms": 8.71199369430542,
"memory_used_mb": 0.0,
"throughput_gb_s": 120.60083660147403,
"tflops": null,
"input_shape": "[1, 1, 4096]",
"model_size_mb": 1050.673728
},
{
"model_name": "DeepHermes_lm_head",
"backend": "ANE",
"inference_time_ms": 8.70786984761556,
"memory_used_mb": 0.0,
"throughput_gb_s": 120.65795038125215,
"tflops": null,
"input_shape": "[1, 1, 4096]",
"model_size_mb": 1050.673728
},
{
"model_name": "llama_lm_head (Parallel Run)",
"backend": "ANE",
"inference_time_ms": 17.02842950820923,
"memory_used_mb": 0.0,
"throughput_gb_s": 61.701152622059546,
"tflops": null,
"input_shape": "[1, 1, 4096]",
"model_size_mb": 1050.673728
},
{
"model_name": "DeepHermes_lm_head (Parallel Run)",
"backend": "ANE",
"inference_time_ms": 17.346296310424805,
"memory_used_mb": 0.0,
"throughput_gb_s": 60.570493504631564,
"tflops": null,
"input_shape": "[1, 1, 4096]",
"model_size_mb": 1050.673728
},
{
"model_name": "Combined (llama_lm_head + DeepHermes_lm_head)",
"backend": "ANE",
"inference_time_ms": 17.346296310424805,
"memory_used_mb": 0.0,
"throughput_gb_s": 120.93652731402324,
"tflops": null,
"input_shape": "[]",
"model_size_mb": 2101.347456
}
]
}

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions