Qwen-2.5-1B-RLCD-Fast / mac_cpu_continuation.json
epsilon3's picture
Add tree-attention fast path and CUDA/M4 benchmarks
f36843f verified
Raw History Blame Contribute Delete
3.19 kB
{
"environment": {
"model": "Qwen/Qwen2.5-1.5B-Instruct",
"revision": "989aa7980e4cf806f80c7fef2b1adb7bc71aa306",
"torch": "2.11.0",
"transformers": "4.57.6",
"gpu": null,
"platform": "macOS-26.6.2-arm64-arm-64bit",
"dtype": "torch.float32",
"attention": "sdpa"
},
"arguments": {
"model": "Qwen/Qwen2.5-1.5B-Instruct",
"presets": [
"fintech_fraud"
],
"fields": [
4
],
"steps": 4,
"repeats": 8,
"warmup": 2,
"output": "mac_cpu_continuation.json"
},
"cases": [
{
"preset": "fintech_fraud",
"fields": 4,
"prefix_tokens": 291,
"suffix_tokens": 28,
"batch_padded_tokens": 32,
"correctness": {
"max_abs_logit_error": 6.008148193359375e-05,
"mean_abs_logit_error": 6.594165370188421e-06,
"full_vocab_argmax_agreement": 1.0,
"candidate_decisions_equal": true,
"candidate_collision_fields": []
},
"decode": {
"batch": {
"median_ms": 622.972208991996,
"min_ms": 586.4072090043919,
"p90_ms": 801.8490410031518,
"samples_ms": [
586.4072090043919,
600.5369170015911,
603.2724169926951,
680.7952079980168,
644.0831670042826,
614.7183339926414,
631.2260839913506,
801.8490410031518
]
},
"tree": {
"median_ms": 585.8095005023642,
"min_ms": 559.2341670126189,
"p90_ms": 626.3385829952313,
"samples_ms": [
568.0354160140269,
596.4209999947343,
559.2341670126189,
578.7606250087265,
626.3385829952313,
579.8298340087058,
591.7891669960227,
613.8133749918779
]
}
},
"end_to_end": {
"batch": {
"median_ms": 967.1460625031614,
"min_ms": 940.2594160055742,
"p90_ms": 1015.3512500110082,
"samples_ms": [
999.5129159942735,
940.2594160055742,
1015.3512500110082,
962.8962919960031,
971.3958330103196,
977.6666249963455,
943.482041999232,
960.0878339988412
]
},
"tree": {
"median_ms": 938.4278540019295,
"min_ms": 914.2270419979468,
"p90_ms": 957.8748749918304,
"samples_ms": [
928.7943340023048,
914.2270419979468,
957.8748749918304,
952.5329169991892,
935.7932499988237,
941.0624580050353,
943.823458001134,
917.1714999974938
]
}
},
"memory": {},
"decode_speedup": 1.0634382140572365,
"end_to_end_speedup": 1.0306024681372818,
"decode_speedup_ci95": [
1.0187897902697607,
1.1586212607706694
],
"end_to_end_speedup_ci95": [
1.0099125150404291,
1.0610482668353582
]
}
]
}