| { |
| "mlx_version": "0.32.4", |
| "tensor_count": 1899, |
| "parameter_elements": 4312697600, |
| "weight_bytes": 10158060544, |
| "dtypes": { |
| "float32": 1151, |
| "float16": 748 |
| }, |
| "strict_load": { |
| "missing": 0, |
| "extra": 0, |
| "shape_mismatches": 0 |
| }, |
| "regression_tests": { |
| "class": "TestHi3D", |
| "passed": 14 |
| }, |
| "reference_comparison": { |
| "reference": "Original Hi3D PyTorch code (github 23dcc68, hub 705fc09) on CPU; spconv and flash-attn replaced by reference implementations; float32 unless noted", |
| "mlx_device": "GPU", |
| "reports": { |
| "sphere64_report": { |
| "conditioning": { |
| "max_abs": 5.714595317840576e-06, |
| "mean_abs": 2.006490236685449e-07, |
| "rel_l2": 8.875883576616418e-06 |
| }, |
| "sigmas": { |
| "max_abs": 5.960464477539063e-08, |
| "mean_abs": 4.584972675030048e-09, |
| "rel_l2": 2.0996074718879118e-08 |
| }, |
| "times": { |
| "max_abs": 5.960464477539063e-08, |
| "mean_abs": 4.76837158203125e-09, |
| "rel_l2": 2.0855193972367177e-08 |
| }, |
| "velocity_fp16_vs_fp32ref": { |
| "max_abs": 0.013125389814376831, |
| "mean_abs": 0.0010297759424406192, |
| "rel_l2": 0.0012895214320252043 |
| }, |
| "velocity_fp16_vs_fp16ref": { |
| "max_abs": 0.015636205673217773, |
| "mean_abs": 0.001322876022487012, |
| "rel_l2": 0.0016291599890619199 |
| }, |
| "ref_fp16_vs_ref_fp32": { |
| "max_abs": 0.016333699226379395, |
| "mean_abs": 0.0011320557014777956, |
| "rel_l2": 0.0013910033166357426 |
| }, |
| "velocity_fp32": { |
| "max_abs": 2.0503997802734375e-05, |
| "mean_abs": 1.8179455511446113e-06, |
| "rel_l2": 2.220910126120128e-06 |
| }, |
| "decoded_voxels": { |
| "ours": 1442341, |
| "reference": 1442339, |
| "common": 1442333 |
| }, |
| "decoded_sdf_sign": { |
| "max_abs": 28.49630832672119, |
| "mean_abs": 0.000985199632137544, |
| "rel_l2": 0.0035812783661513725 |
| }, |
| "decoded_feat": { |
| "max_abs": 0.3575434908270836, |
| "mean_abs": 2.4584798492266753e-05, |
| "rel_l2": 0.0023846487551618238 |
| }, |
| "mesh_faces_identical": true, |
| "mesh_vertices": { |
| "max_abs": 2.86102294921875e-06, |
| "mean_abs": 2.4002885922643986e-07, |
| "rel_l2": 5.207820769835567e-08 |
| }, |
| "filled_faces_identical": true |
| }, |
| "sphere192_report": { |
| "conditioning": { |
| "max_abs": 5.714595317840576e-06, |
| "mean_abs": 2.006490236685449e-07, |
| "rel_l2": 8.875883576616418e-06 |
| }, |
| "sigmas": { |
| "max_abs": 0.0, |
| "mean_abs": 0.0, |
| "rel_l2": 0.0 |
| }, |
| "times": { |
| "max_abs": 0.0, |
| "mean_abs": 0.0, |
| "rel_l2": 0.0 |
| }, |
| "velocity_fp16_vs_fp32ref": { |
| "max_abs": 0.032924532890319824, |
| "mean_abs": 0.0010041114637135014, |
| "rel_l2": 0.001261798954717289 |
| }, |
| "velocity_fp32": { |
| "max_abs": 5.2988529205322266e-05, |
| "mean_abs": 2.1241101601153427e-06, |
| "rel_l2": 2.638211684786904e-06 |
| } |
| } |
| } |
| }, |
| "benchmark": { |
| "hardware": "Apple M4 Pro, 64 GiB unified memory", |
| "python": "3.12.11", |
| "input": "Hi3D assets/example.png (436x640 RGBA), coarse mesh from mlx-community/sam-3d-objects-bf16 (seed 0)", |
| "resolution": 1024, |
| "steps": 25, |
| "seed": 0, |
| "timings": { |
| "coarse_mesh": 77.6, |
| "load": 1.5, |
| "conditioning": 0.58, |
| "pass1_sampling": 189.08, |
| "pass1_decoding": 8.86, |
| "pass2_sampling": 1517.58, |
| "pass2_decoding": 44.96, |
| "postprocess": 8.67 |
| }, |
| "peak": 30.05, |
| "vertices": 495425, |
| "faces": 1000000 |
| } |
| } |
|
|
