| { |
| "fixtureArrays": { |
| "topk_small_rows_rank3_cycle33": [0.015625, 0.03125, 0.046875, 0.0625, 0.078125, 0.09375, 0.109375, 0.125, 0.140625, 0.15625, 0.171875, 0.1875, 0.203125, 0.21875, 0.234375, 0.25, 0.265625, 0.28125, 0.296875, 0.3125, 0.328125, 0.34375, 0.359375, 0.375, 0.390625, 0.40625, 0.421875, 0.4375, 0.453125, 0.46875, 0.484375, 0.5, 0.515625], |
| "order_key_extremes": ["-Infinity", -65504, -2, 0, 0, 2, 65504, "Infinity", 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0] |
| }, |
| "cases": [ |
| { |
| "name": "onnx_default_last_axis_omitted", |
| "provenance": { |
| "source": "https://onnx.ai/onnx/operators/onnx__TopK.html", |
| "test": "TopK axis default", |
| "notes": "Omitting axis must select the last dimension (ONNX default -1), not axis 1. Rank 3 makes those axes distinct, and both output values and indices are pinned. Representable ONNX int64 indices are stored in uint32 slots." |
| }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 2, 3], |
| "data": { "kind": "values", "values": [1.0, 9.0, 3.0, 4.0, 2.0, 8.0, 7.0, 0.0, 6.0, 5.0, 11.0, 10.0] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "float32", |
| "shape": [2, 2, 1], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [9.0, 8.0, 7.0, 11.0] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [2, 2, 1], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [1, 2, 0, 1] } |
| } |
| } |
| }, |
| { |
| "name": "f32_largest_positive_subnormal_beats_zero", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top1DefaultAxisLargestElements", |
| "notes": "A positive subnormal is strictly greater than zero; TopK should return its index rather than tie-breaking on a flushed zero." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [1, 3], "data": { "kind": "values", "values": [-1e-40, 0.0, 1e-40] } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 1], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "f32_largest_positive_subnormal_beats_zero_axis0", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top1DefaultAxisLargestElements", |
| "notes": "On axis 0, a positive subnormal is strictly greater than zero; TopK must return its index rather than tie-break against a flushed zero." |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [3, 1], "data": { "kind": "values", "values": [-1e-40, 0.0, 1e-40] } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 1], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "f32_smallest_negative_subnormal_beats_zero", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top2DefaultAxisSmallestElements", |
| "notes": "A negative subnormal is strictly less than zero; smallest TopK should return its index rather than tie-breaking on flushed zeros." |
| }, |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [1, 3], "data": { "kind": "values", "values": [1e-40, 0.0, -1e-40] } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 1], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 1], "tolerance": 0, "data": { "kind": "values", "values": [2] } } |
| } |
| }, |
| { |
| "name": "f32_smallest_negative_subnormal_beats_zero_axis0", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3ExplicitAxisSmallestElements", |
| "notes": "On axis 0, a negative subnormal is strictly less than zero; smallest TopK must return its index." |
| }, |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [3, 1], "data": { "kind": "values", "values": [1e-40, 0.0, -1e-40] } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 1], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 1], "tolerance": 0, "data": { "kind": "values", "values": [2] } } |
| } |
| }, |
| { |
| "name": "axis1_largest_sorted", |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 5], |
| "data": { "kind": "values", "values": [1.5, 2.0, 2.0, -1.0, 0.5, 0.1, 2.5, 3.0, 3.0, -2.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 1], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top1_default_axis_largest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top1DefaultAxisLargestElements" |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [0.1, 0.3, 0.2, 0.4, 0.1, 0.3, 0.3, 0.2] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 1], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top2_default_axis_largest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top2DefaultAxisLargestElements" |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [0.1, 0.3, 0.2, 0.4, 0.1, 0.3, 0.4, 0.2] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top3_default_axis_largest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3DefaultAxisLargestElements" |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [0.1, 0.3, 0.2, 0.4, 0.1, 0.3, 0.4, 0.2] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top_all_default_axis_largest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopAllDefaultAxisLargestElements" |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [0.1, 0.3, 0.2, 0.4, 0.1, 0.3, 0.3, 0.2] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 4], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top2_default_axis_smallest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top2DefaultAxisSmallestElements" |
| }, |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [0.1, 0.3, 0.2, 0.4, 0.1, 0.3, 0.4, 0.2] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top3_axis0_smallest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3ExplicitAxisSmallestElements" |
| }, |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { "kind": "values", "values": [0.1, 0.3, 0.2, 0.4, 0.1, 0.3, 0.3, 0.2] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [3, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top1_rank3_axis1_largest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top1ExplicitAxisMultiDInputLargestElements" |
| }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 2, 2], |
| "data": { "kind": "values", "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 1, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 1, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top1_rank3_axis1_smallest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top1ExplicitAxisMultiDInputSmallestElements" |
| }, |
| "attrs": { "axis": 1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 2, 2], |
| "data": { "kind": "values", "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 1, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 1, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top2_axis0_largest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top2ExplicitAxisLargestElements" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 4], |
| "data": { "kind": "values", "values": [0.0, 1.0, 2.0, 11.0, 8.0, 5.0, 6.0, 7.0, 4.0, 9.0, 10.0, 3.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 4], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top_all_axis0_largest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopAllExplicitAxisLargestElements" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { "kind": "values", "values": [0.1, 0.3, 0.2, 0.4, 0.1, 0.3, 0.3, 0.2] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [4, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [4, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "tie_breaks_lower_index", |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 6], |
| "data": { "kind": "values", "values": [5.0, 5.0, 4.0, 5.0, 3.0, 5.0, 1.0, 2.0, 2.0, 2.0, -1.0, -1.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 4], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "smallest_tie_breaks_lower_index", |
| "attrs": { "axis": 1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 6], |
| "data": { "kind": "values", "values": [-5.0, -5.0, -4.0, -5.0, 3.0, -5.0, 1.0, -2.0, -2.0, -2.0, -1.0, -1.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "axis0_largest_sorted", |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 4], |
| "data": { "kind": "values", "values": [1.0, 5.0, 3.0, 4.0, 2.0, 5.0, 7.0, 1.0, 3.0, 4.0, 7.0, 9.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 4], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "axis0_smallest_sorted", |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 4], |
| "data": { "kind": "values", "values": [1.0, 5.0, 3.0, 4.0, 2.0, 5.0, -7.0, 1.0, 3.0, 4.0, -7.0, 9.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 4], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "axis_minus_one_largest_sorted", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 5], |
| "data": { "kind": "values", "values": [0.5, 4.0, 4.0, -1.0, 2.0, 7.0, 1.0, 7.0, 3.0, 7.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "axis1_largest_unsorted_attr", |
| "attrs": { "axis": 1, "largest": 1, "sorted": 0 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 5], |
| "data": { "kind": "values", "values": [1.0, 5.0, 3.0, 4.0, 2.0, 2.0, 7.0, 8.0, 6.0, 1.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 1], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "axis_minus_two_largest_sorted", |
| "attrs": { "axis": -2, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 4], |
| "data": { "kind": "values", "values": [1.0, 5.0, 3.0, 4.0, 2.0, 5.0, 7.0, 1.0, 3.0, 4.0, 7.0, 9.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 4], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_last_axis_largest_sorted_ties", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 2, 5], |
| "data": { |
| "kind": "values", |
| "values": [1.0, 4.0, 4.0, -2.0, 3.0, 0.0, -1.0, -1.0, 5.0, 5.0, 9.0, 8.0, 9.0, 7.0, 6.0, -3.0, -3.0, -4.0, -2.0, -2.0] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 2, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_rank3_last_axis_largest_nan_rows", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top2DefaultAxisLargestElements", |
| "notes": "NaN extension of ORT's largest sorted TopK coverage. Includes an all-NaN row and a finite-tie row where a NaN sentinel appears after finite values." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [1, 2, 4], |
| "data": { "kind": "values", "values": ["NaN", "NaN", "NaN", "NaN", 1.0, "NaN", 1.0, 0.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 2, 2], "tolerance": 0, "allowNaN": true }, |
| "indices": { "dtype": "uint32", "shape": [1, 2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_last_axis_smallest_sorted_ties", |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 2, 5], |
| "data": { |
| "kind": "values", |
| "values": [1.0, -4.0, -4.0, 2.0, -3.0, 0.0, -1.0, -1.0, -5.0, -5.0, -9.0, -8.0, -9.0, -7.0, -6.0, 3.0, 3.0, 4.0, 2.0, 2.0] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 2, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis1_largest_sorted_ties", |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 4, 3], |
| "data": { |
| "kind": "values", |
| "values": [1.0, 5.0, 2.0, 3.0, 5.0, 4.0, 3.0, -1.0, 6.0, 0.0, 7.0, 6.0, -2.0, 10.0, 0.0, -2.0, 8.0, 0.0, 4.0, 8.0, -1.0, 4.0, 9.0, -1.0] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 2, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis1_smallest_sorted_ties", |
| "attrs": { "axis": 1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 4, 3], |
| "data": { |
| "kind": "values", |
| "values": [-1.0, 5.0, -2.0, -3.0, 5.0, -4.0, -3.0, -1.0, -6.0, 0.0, 7.0, -6.0, 2.0, -10.0, 0.0, 2.0, -8.0, 0.0, -4.0, -8.0, 1.0, -4.0, -9.0, 1.0] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 2, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis0_largest_sorted_ties", |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 2, 3], |
| "data": { |
| "kind": "values", |
| "values": [1.0, 9.0, 3.0, 4.0, 5.0, 6.0, 2.0, 9.0, 7.0, 4.0, 8.0, 6.0, 3.0, 1.0, 7.0, 2.0, 8.0, 0.0, 3.0, 9.0, -1.0, 4.0, 0.0, 6.0] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 2, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis0_smallest_sorted_ties", |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 2, 3], |
| "data": { |
| "kind": "values", |
| "values": [-1.0, -9.0, 3.0, 4.0, -5.0, -6.0, -2.0, -9.0, -7.0, 4.0, -8.0, -6.0, -3.0, -1.0, -7.0, 2.0, -8.0, 0.0, -3.0, -9.0, -1.0, 4.0, 0.0, -6.0] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 2, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank4_last_axis_largest_sorted_ties", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [1, 2, 2, 4], |
| "data": { |
| "kind": "values", |
| "values": [1.0, 4.0, 4.0, 2.0, 0.0, 5.0, 5.0, 3.0, 7.0, 6.0, 7.0, 1.0, -2.0, -2.0, -3.0, -1.0] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 2, 2, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 2, 2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank4_last_axis_smallest_sorted_ties", |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [1, 2, 2, 4], |
| "data": { |
| "kind": "values", |
| "values": [-1.0, -4.0, -4.0, -2.0, 0.0, -5.0, -5.0, -3.0, -7.0, -6.0, -7.0, -1.0, 2.0, 2.0, 3.0, 1.0] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 2, 2, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 2, 2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "top0_default_axis_largest_empty_outputs", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top0DefaultAxisLargestElements" |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 0 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [0.1, 0.3, 0.2, 0.4, 0.1, 0.3, 0.3, 0.2] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 0], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 0], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "top0_axis0_empty_outputs", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top0DefaultAxisLargestElements" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 0 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { "kind": "values", "values": [0.1, 0.3, 0.2, 0.4, 0.1, 0.3, 0.3, 0.2] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [0, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [0, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank1_top_all_largest_ort", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopAllExplicitAxis1DInputLargestElements" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 13 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [13], |
| "data": { |
| "kind": "values", |
| "values": [93.0, 695.0, 971.0, 978.0, 483.0, 247.0, 242.0, 983.0, 531.0, 723.0, 285.0, 527.0, 862.0] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [13], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [13], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank1_top2_repeated_maxima_lower_index", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopAllExplicitAxis1DLargeInputLargestElements" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [16], |
| "data": { |
| "kind": "values", |
| "values": [93.0, 695.0, 971.0, 978.0, 483.0, 247.0, 242.0, 983.0, 531.0, 723.0, 93.0, 695.0, 971.0, 983.0, 483.0, 247.0] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "f16_rank1_nth_element_negative_values", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.NthElementHalf_NegtiveVals" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [6], |
| "data": { "kind": "values", "values": [10.0, -8.0, -7.0, -4.0, -5.0, -6.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [4], "tolerance": 0.00098 }, |
| "indices": { "dtype": "uint32", "shape": [4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "smallest_non_power_of_two_positive_infinity_padding", |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [1, 3], "data": { "kind": "values", "values": [1.0, 2.0, "Infinity"] } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "largest_non_power_of_two_negative_infinity_padding", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [1, 3], "data": { "kind": "values", "values": [-1.0, -2.0, "-Infinity"] } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "top3_all_same_smallest_lower_indices", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3AllSame" |
| }, |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 0.1] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_axis0_all_positive_infinity_largest_ties", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3ExplicitAxisInfinity" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { |
| "kind": "values", |
| "values": ["Infinity", "Infinity", "Infinity", "Infinity", "Infinity", "Infinity", "Infinity", "Infinity"] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [3, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_axis0_all_negative_infinity_largest_ties", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3ExplicitAxisInfinity" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { |
| "kind": "values", |
| "values": ["-Infinity", "-Infinity", "-Infinity", "-Infinity", "-Infinity", "-Infinity", "-Infinity", "-Infinity"] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [3, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_rank1_one_merge_all_ties_largest", |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 8 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [4097], "data": { "kind": "constant", "value": 0.25 } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [8], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [8], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_axis1_one_merge_all_ties_smallest", |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 7 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [2, 4099], "data": { "kind": "constant", "value": -0.5 } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 7], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 7], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_axis1_two_merge_vocab_top100_ties", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 100 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [1, 43009], "data": { "kind": "constant", "value": 1.0 } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 100], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 100], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_axis1_two_merge_smallest_ties", |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 100 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [1, 43009], "data": { "kind": "constant", "value": -1.0 } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 100], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 100], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_f16_one_merge_generated_logits", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 6 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [1, 3000], |
| "data": { "kind": "fillFloat32", "sinStep": 0.013, "cosStep": 0.029, "scale": 2.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [1, 6], "tolerance": 0.00098 }, |
| "indices": { "dtype": "uint32", "shape": [1, 6], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_top0_last_axis_empty_outputs", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 0 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [2, 4097], "data": { "kind": "constant", "value": 0.0 } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 0], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 0], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_top1_last_axis_all_ties_lower_index", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 0 }, |
| "args": { "k": 1 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [1, 4097], "data": { "kind": "constant", "value": 3.0 } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 1], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_top1_last_axis_smallest_generated", |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 3001], |
| "data": { "kind": "fillFloat32", "sinStep": 0.017, "cosStep": 0.031, "scale": 2.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 1], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "last_axis_top1_smallest_generated", |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 129], |
| "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.043, "scale": 2.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 1], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [3, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank1_axis0_smallest_sorted_tie_lower_index", |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [7], |
| "data": { "kind": "values", "values": [3.0, -1.5, 2.0, -1.5, 0.0, 5.0, -4.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "f16_axis0_non_last_all_ties", |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { "x": { "dtype": "float16", "shape": [6, 4], "data": { "kind": "constant", "value": 2.0 } } }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [2, 4], "tolerance": 0.00098 }, |
| "indices": { "dtype": "uint32", "shape": [2, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis1_unsorted_top1_smallest", |
| "attrs": { "axis": 1, "largest": 0, "sorted": 0 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 5, 3], |
| "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.07, "scale": 1.5 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 1, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 1, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_axis0_top3_positive_infinity_smallest", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3ExplicitAxisInfinity" |
| }, |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { |
| "kind": "values", |
| "values": ["Infinity", "Infinity", "Infinity", "Infinity", "Infinity", "Infinity", "Infinity", "Infinity"] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 2], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_axis0_top3_negative_infinity_smallest", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3ExplicitAxisInfinity" |
| }, |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { |
| "kind": "values", |
| "values": ["-Infinity", "-Infinity", "-Infinity", "-Infinity", "-Infinity", "-Infinity", "-Infinity", "-Infinity"] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 2], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_rank1_nth_element_k4_f32", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.NthElement" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [6], |
| "data": { "kind": "values", "values": [10.0, 8.0, 7.0, 4.0, 5.0, 6.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [4], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_rank1_nth_element_k4_negative_values", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.NthElementHalf_NegtiveVals" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [6], |
| "data": { "kind": "values", "values": [10.0, -8.0, -7.0, -4.0, -5.0, -6.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [4], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_rank1_nth_element_k4_f16", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.NthElementHalf" |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [6], |
| "data": { "kind": "values", "values": [10.0, 8.0, 7.0, 4.0, 5.0, 6.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [4], "tolerance": 0.00098 }, |
| "indices": { "dtype": "uint32", "shape": [4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank1_axis0_largest_k100_merge", |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 100 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [1024], |
| "data": { "kind": "fillFloat32", "sinStep": 0.017, "cosStep": 0.031, "scale": 2.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [100], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [100], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_top3_all_same_smallest_tie_lower_index", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3AllSame" |
| }, |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [2, 4], "data": { "kind": "constant", "value": 0.1 } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 3], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "onnx_backend_top_k", |
| "attrs": { "axis": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 4], |
| "data": { "kind": "values", "values": [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 3] }, |
| "indices": { "dtype": "uint32", "shape": [3, 3] } |
| }, |
| "provenance": { |
| "source": "cmake/external/onnx/onnx/backend/test/data/node/test_top_k", |
| "notes": "The fixture represents the scalar TopK `K` input as argument `k`. This WebGPU package stores representable ONNX int64 values and indices in int32 or uint32 slots." |
| } |
| }, |
| { |
| "name": "onnx_backend_top_k_negative_axis", |
| "attrs": { "axis": -1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 4], |
| "data": { "kind": "values", "values": [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 3] }, |
| "indices": { "dtype": "uint32", "shape": [3, 3] } |
| }, |
| "provenance": { |
| "source": "cmake/external/onnx/onnx/backend/test/data/node/test_top_k_negative_axis", |
| "notes": "The fixture represents the scalar TopK `K` input as argument `k`. This WebGPU package stores representable ONNX int64 values and indices in int32 or uint32 slots." |
| } |
| }, |
| { |
| "name": "onnx_backend_top_k_smallest", |
| "attrs": { "axis": 1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 4], |
| "data": { "kind": "values", "values": [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 11.0, 10.0, 9.0, 8.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 3] }, |
| "indices": { "dtype": "uint32", "shape": [3, 3] } |
| }, |
| "provenance": { |
| "source": "cmake/external/onnx/onnx/backend/test/data/node/test_top_k_smallest", |
| "notes": "The fixture represents the scalar TopK `K` input as argument `k`. This WebGPU package stores representable ONNX int64 values and indices in int32 or uint32 slots." |
| } |
| }, |
| { |
| "name": "onnx_backend_top_k_same_values_2d", |
| "provenance": { |
| "source": "cmake/external/onnx/onnx/backend/test/data/node/test_top_k_same_values_2d", |
| "notes": "Official value tensor is int64; adapted to float32 because this TopK manifest supports floating values and uint32 indices." |
| }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 4], |
| "data": { "kind": "values", "values": [0.0, 0.0, 0.0, 0.0, 1.0, 1.0, 1.0, 1.0, 2.0, 2.0, 1.0, 1.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 3], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "onnx_backend_top_k_same_values_largest", |
| "provenance": { |
| "source": "cmake/external/onnx/onnx/backend/test/data/node/test_top_k_same_values_largest", |
| "notes": "The official int64 value tensor is represented as supported float32 values with uint32 indices and an explicit `largest=1` request." |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [4], "data": { "kind": "values", "values": [0.0, 0.0, 0.0, 0.0] } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_int8_largest_sorted_tie_indices", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopK_Int8" |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "int8", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [10, 30, 20, 40, 10, 30, 40, 20] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "int8", "shape": [2, 2], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_int8_negative_largest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopK_Int8_Negative" |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "int8", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [-10, -30, -20, -40, -10, -30, -40, -20] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "int8", "shape": [2, 2], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_int8_smallest_sorted", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopK_Int8_Smallest" |
| }, |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "int8", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [10, 30, 20, 40, 10, 30, 40, 20] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "int8", "shape": [2, 2], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_int16_largest_sorted_tie_indices", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopK_Int16" |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "int16", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [100, 300, 200, 400, 100, 300, 400, 200] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "int16", "shape": [2, 2], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_uint8_largest_sorted_tie_indices", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopK_Uint8" |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "uint8", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [10, 30, 20, 40, 10, 30, 40, 20] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "uint8", "shape": [2, 2], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "int32_largest_sorted_tie_indices", |
| "provenance": { |
| "source": "https://github.com/onnx/onnx/blob/main/docs/Operators.md#TopK", |
| "notes": "ONNX TopK permits int32 value tensors." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [10, 30, 20, 40, 40, 30, 40, 20] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "int32", "shape": [2, 2], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "int32_largest_exact_above_float24", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top1ExplicitAxisMultiDInputLargestElements", |
| "notes": "Distinct int32 values 16777216 and 16777217 must remain ordered by integer value, not by their rounded f32 representation." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [16777216, 16777217, 5, -3, 123456788, 123456789, 0, -7] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [2, 1], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [16777217, 123456789] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [2, 1], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [1, 1] } |
| } |
| } |
| }, |
| { |
| "name": "int32_smallest_exact_below_negative_float24", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3ExplicitAxisSmallestElements", |
| "notes": "Distinct int32 values -16777216 and -16777217 must remain ordered by integer value, not by their rounded f32 representation." |
| }, |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [-16777216, -16777217, 5, -3, -123456788, -123456789, 0, -7] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [2, 1], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [-16777217, -123456789] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [2, 1], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [1, 1] } |
| } |
| } |
| }, |
| { |
| "name": "axis0_int32_largest_exact_above_float24", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top1ExplicitAxisLargestElements", |
| "notes": "An axis-0 TopK must compare and emit exact int32 values without routing them through f32." |
| }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 2], |
| "data": { "kind": "values", "values": [16777216, 123456789, 16777217, 123456788] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [1, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [16777217, 123456789] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [1, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [1, 0] } |
| } |
| } |
| }, |
| { |
| "name": "axis0_int32_smallest_exact_below_negative_float24", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top3ExplicitAxisSmallestElements", |
| "notes": "An axis-0 smallest TopK must preserve exact int32 ordering below -2^24; an f32 round trip collapses the chosen sentinel values." |
| }, |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 2], |
| "data": { "kind": "values", "values": [-16777216, -123456789, -16777217, -123456788] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [1, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [-16777217, -123456789] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [1, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [1, 0] } |
| } |
| } |
| }, |
| { |
| "name": "int32_largest_top2_exact_above_float24", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top2ExplicitAxisMultiDInputLargestElements", |
| "notes": "Sorted k=2 must order exact int32 values before applying tie-breaking; an f32 comparison would see 16777216 and 16777217 as equal." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [16777216, 16777217, 5, -3, 123456788, 123456789, 0, -7] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [2, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [16777217, 16777216, 123456789, 123456788] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [2, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [1, 0, 1, 0] } |
| } |
| } |
| }, |
| { |
| "name": "int32_smallest_top2_exact_below_negative_float24", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.Top2ExplicitAxisMultiDInputSmallestElements", |
| "notes": "Sorted smallest TopK with k=2 must preserve exact int32 ordering below -2^24." |
| }, |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 4], |
| "data": { "kind": "values", "values": [-16777216, -16777217, 5, -3, -123456788, -123456789, 0, -7] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [2, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [-16777217, -16777216, -123456789, -123456788] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [2, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [1, 0, 1, 0] } |
| } |
| } |
| }, |
| { |
| "name": "rank3_axis1_int32_largest_exact_above_float24", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopK_Int32_ExplicitAxis", |
| "notes": "Middle-axis TopK must use exact int32 ordering; this covers the rank-3 axis-1 path as well as the last-axis and axis-0 paths." |
| }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [1, 2, 2], |
| "data": { "kind": "values", "values": [16777216, 123456789, 16777217, 123456788] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [1, 1, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [16777217, 123456789] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [1, 1, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [1, 0] } |
| } |
| } |
| }, |
| { |
| "name": "rank3_axis1_int32_smallest_exact_below_negative_float24", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopK_Int32_ExplicitAxis", |
| "notes": "A rank-3 middle-axis smallest TopK must preserve exact int32 ordering below -2^24." |
| }, |
| "attrs": { "axis": 1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [1, 2, 2], |
| "data": { "kind": "values", "values": [-16777216, -123456789, -16777217, -123456788] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [1, 1, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [-16777217, -123456789] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [1, 1, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [1, 0] } |
| } |
| } |
| }, |
| { |
| "name": "ort_int8_explicit_axis_middle", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.TopK_Int8_ExplicitAxis" |
| }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { "dtype": "int8", "shape": [2, 2, 2], "data": { "kind": "values", "values": [1, 2, 3, 4, 5, 6, 7, 8] } } |
| }, |
| "outputs": { |
| "values": { "dtype": "int8", "shape": [2, 1, 2], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 1, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "dispatch_cliff_rows_65537", |
| "provenance": { |
| "notes": "With 65,537 rows, TopK uses a two-row workgroup dispatch. An 11-value cycle shifts every eight-element row and avoids ties, making folded row-index errors observable." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [65537, 8], |
| "data": { "kind": "cycle", "values": [0.1, 0.5, 0.9, 0.3, 0.7, 0.2, 0.6, 0.4, 0.8, 0.05, 0.95] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [65537, 4], "tolerance": 0.00001 }, |
| "indices": { "dtype": "uint32", "shape": [65537, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "dispatch_cliff_rank3_axis1_cols_65537", |
| "provenance": { |
| "notes": "A rank-3 axis-1 TopK with trailing extent 65,537 uses a two-row workgroup dispatch while the z dimension carries the batch. Distinct column values make folded column-index errors observable." |
| }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 2, 65537], |
| "data": { "kind": "cycle", "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 1, 65537], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 1, 65537], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "f16_vocab_top1_large_block_argmax", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [1, 4096], |
| "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.027, "scale": 3.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [1, 1], "tolerance": 0.001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "axis0_largest_k8_rows4096_tournament", |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "args": { "k": 8 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4096, 8], |
| "data": { "kind": "fillFloat32", "sinStep": 0.013, "cosStep": 0.037, "scale": 2.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [8, 8], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [8, 8], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "axis1_empty_row_dim_largest", |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [0, 6], "data": { "kind": "constant", "value": 1.0 } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [0, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [0, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "axis1_k_equals_cols_minus_one_pad_vs_real_zero", |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2046 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 2047], |
| "data": { "kind": "fillFloat32", "sinStep": 0.017, "cosStep": 0.023, "scale": 0.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 2046], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [2, 2046], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_merge_last_axis_pos_inf_ties_lower_index", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 5 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 4100], |
| "data": { |
| "kind": "cycle", |
| "values": ["Infinity", 1.0, 2.0, "Infinity", 3.0, "Infinity", 0.5, 4.0, "Infinity", 2.5] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 5], "tolerance": 0, "allowNaN": true }, |
| "indices": { "dtype": "uint32", "shape": [2, 5], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "small_row_batched_boundary_4097x8_k4", |
| "provenance": { |
| "notes": "The 4,097th eight-element row checks the final partial row group; an 11-value cycle exercises row-dependent ordering and ties." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4097, 8], |
| "data": { "kind": "cycle", "values": [0.1, 0.5, 0.9, 0.3, 0.7, 0.2, 0.6, 0.4, 0.8, 0.05, 0.95] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [4097, 4], "tolerance": 0.00001 }, |
| "indices": { "dtype": "uint32", "shape": [4097, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rows_smallk_boundary_1025x128_k4", |
| "provenance": { |
| "notes": "The 1,025th row checks partial-row handling, stable ties and the first-index ordering for k=4." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [1025, 128], |
| "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5, "srqStep": 0.02 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1025, 4], "tolerance": 0.00001 }, |
| "indices": { "dtype": "uint32", "shape": [1025, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "small_row_batched_smallest_ties_4096x8_k4", |
| "provenance": { |
| "notes": "For 4,096 equal eight-element rows with largest=0, sorted output must retain the lower input indices in ascending order." |
| }, |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [4096, 8], "data": { "kind": "constant", "value": -1.0 } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [4096, 4], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [4096, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis1_smallk_tournament_16x2048x16_k8", |
| "provenance": { |
| "notes": "A 2048-element axis across 256 independent rows checks sorted top-eight values and indices." |
| }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 8 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [16, 2048, 16], |
| "data": { "kind": "fillFloat32", "sinStep": 0.013, "cosStep": 0.037, "scale": 2.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [16, 8, 16], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [16, 8, 16], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis1_k9_256x2048x1_tournament_boundary", |
| "provenance": { |
| "notes": "A [256, 2048, 1] float32 tensor holds a deterministic sinusoidal fill; TopK keeps the 9 largest values (largest=1, sorted=1) along axis 1 independently for each of the 256 rows." |
| }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 9 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [256, 2048, 1], |
| "data": { "kind": "fillFloat32", "sinStep": 0.017, "cosStep": 0.031, "scale": 2.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [256, 9, 1], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [256, 9, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis1_2049x_k8_tournament_coverage", |
| "provenance": { |
| "notes": "A non-last axis of length 2,049 gives each tournament lane nine local candidates and avoids the generic bitonic implementation." |
| }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 8 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [256, 2049, 1], |
| "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.029, "scale": 2.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [256, 8, 1], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [256, 8, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_int32_top1_preserves_exact_order", |
| "provenance": { |
| "notes": "Large-axis integer top-1 must use the native direct reducer: f32 split scratch would collapse 16777216 and 16777217 to the same value." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 1 }, |
| "inputs": { |
| "x": { "dtype": "int32", "shape": [1, 2049], "data": { "kind": "cycle", "values": [16777216, 16777217] } } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [1, 1], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [16777217] } |
| }, |
| "indices": { "dtype": "uint32", "shape": [1, 1], "tolerance": 0, "data": { "kind": "values", "values": [1] } } |
| } |
| }, |
| { |
| "name": "rank5_last_axis_generic_geometry", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 1, 2, 3, 4], |
| "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.29, "scale": 2.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 1, 2, 3, 2], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 1, 2, 3, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "int32_large_axis_merge_top2_above_float24", |
| "provenance": { |
| "source": "onnxruntime/core/providers/webgpu/math/top_k.cc", |
| "test": "float-typed kernel registration", |
| "notes": "Large-axis merge scratch must preserve exact int32 values above 2^24 rather than store candidates as f32." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [3, 2100], |
| "data": { "kind": "cycle", "values": [16777217, -16777217, 5, 16777216, -3, 0, 7, 16777215] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [3, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [16777217, 16777217, 16777217, 16777217, 16777217, 16777217] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [3, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [0, 8, 4, 12, 0, 8] } |
| } |
| } |
| }, |
| { |
| "name": "int32_large_axis_merge_smallest_below_negative_float24", |
| "provenance": { |
| "source": "onnxruntime/core/providers/webgpu/math/top_k.cc", |
| "test": "float-typed kernel registration", |
| "notes": "Large-axis merge scratch must preserve exact negative int32 values below -2^24 rather than store candidates as f32." |
| }, |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 2100], |
| "data": { "kind": "cycle", "values": [-16777217, 16777217, -16777216, 4, -1] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [2, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [-16777217, -16777217, -16777217, -16777217] } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [2, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [0, 5, 0, 5] } |
| } |
| } |
| }, |
| { |
| "name": "uint32_large_axis_merge_top3_above_2pow31", |
| "provenance": { |
| "source": "onnxruntime/core/providers/webgpu/math/top_k.cc", |
| "test": "float-typed kernel registration", |
| "notes": "Large-axis merge scratch must preserve exact uint32 values above 2^31 rather than store candidates as f32." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 3 }, |
| "inputs": { |
| "x": { |
| "dtype": "uint32", |
| "shape": [2, 2100], |
| "data": { "kind": "cycle", "values": [4294967295, 2147483648, 4294967294, 7, 0] } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "uint32", |
| "shape": [2, 3], |
| "tolerance": 0, |
| "data": { |
| "kind": "values", |
| "values": [4294967295, 4294967295, 4294967295, 4294967295, 4294967295, 4294967295] |
| } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [2, 3], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [0, 5, 10, 0, 5, 10] } |
| } |
| } |
| }, |
| { |
| "name": "int32_large_axis_two_merge_k64_above_float24", |
| "provenance": { |
| "source": "onnxruntime/core/providers/webgpu/math/top_k.cc", |
| "test": "float-typed kernel registration", |
| "notes": "Two merge stages must preserve exact int32 values above 2^24 rather than store candidates as f32." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 64 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [1, 20480], |
| "data": { |
| "kind": "cycle", |
| "values": [16777217, 16777219, -16777217, 16777216, 3, 16777221, -5, 16777215, 0, 16777223] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { |
| "dtype": "int32", |
| "shape": [1, 64], |
| "tolerance": 0, |
| "data": { |
| "kind": "values", |
| "values": [16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223, 16777223] |
| } |
| }, |
| "indices": { |
| "dtype": "uint32", |
| "shape": [1, 64], |
| "tolerance": 0, |
| "data": { |
| "kind": "values", |
| "values": [9, 19, 29, 39, 49, 59, 69, 79, 89, 99, 109, 119, 129, 139, 149, 159, 169, 179, 189, 199, 209, 219, 229, 239, 249, 259, 269, 279, 289, 299, 309, 319, 329, 339, 349, 359, 369, 379, 389, 399, 409, 419, 429, 439, 449, 459, 469, 479, 489, 499, 509, 519, 529, 539, 549, 559, 569, 579, 589, 599, 609, 619, 629, 639] |
| } |
| } |
| } |
| }, |
| { |
| "name": "large_vocab_three_merge_k100_262144", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 100 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [1, 262144], |
| "data": { "kind": "fillFloat32", "sinStep": 0.0137, "cosStep": 0.0071, "scale": 4.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 100], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 100], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "large_vocab_three_merge_k200_262144", |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 200 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [1, 262144], |
| "data": { "kind": "fillFloat32", "sinStep": 0.0113, "cosStep": 0.0059, "scale": 4.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 200], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 200], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis1_large_axis8192_k8", |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 8 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 8192, 4], |
| "data": { "kind": "fillFloat32", "sinStep": 0.0173, "cosStep": 0.0091, "scale": 4.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [4, 8, 4], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [4, 8, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis1_large_k32_6144", |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 32 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 6144, 8], |
| "data": { "kind": "fillFloat32", "sinStep": 0.0157, "cosStep": 0.0083, "scale": 4.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 32, 8], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 32, 8], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis1_large_smallest_axis8192_k8", |
| "attrs": { "axis": 1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 8 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4, 8192, 4], |
| "data": { "kind": "fillFloat32", "sinStep": 0.0173, "cosStep": 0.0091, "scale": 4.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [4, 8, 4], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [4, 8, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "two_merge_distinct_values_vocab_k16_262144", |
| "provenance": { |
| "notes": "Distinct values over 262,144 candidates require two merge stages. The final stage must read the immediately preceding candidate buffer; otherwise the returned top 16 values and indices change." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 16 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [1, 262144], |
| "data": { "kind": "fillFloat32", "sinStep": 0.0137, "cosStep": 0.0071, "scale": 4.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 16], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 16], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "two_merge_monotonic_ramp_k128_8704", |
| "provenance": { |
| "notes": "For a monotonic ramp, the true top 128 values occupy the final 128 indices, making stale candidate-buffer reads observable at every merge stage." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 128 }, |
| "inputs": { |
| "x": { "dtype": "float32", "shape": [1, 8704], "data": { "kind": "linspace", "start": 0.0, "end": 1.0 } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 128], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 128], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank3_axis1_two_merge_k128_8704", |
| "provenance": { |
| "notes": "Non-last-axis strided merge with firstCandidates 17*128=2176 past the one-merge chunk, so the two-level strided pipeline runs." |
| }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 128 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [1, 8704, 4], |
| "data": { "kind": "fillFloat32", "sinStep": 0.0157, "cosStep": 0.0083, "scale": 4.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 128, 4], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 128, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "int32_vocab_three_merge_k100_262144", |
| "provenance": { |
| "notes": "Distinct int32 ramp through the three-level merge (512 blocks -> 25 -> 2 chunks); exact order keys keep values above float24 precision honest." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 100 }, |
| "inputs": { |
| "x": { "dtype": "int32", "shape": [1, 262144], "data": { "kind": "linspace", "start": -131072, "end": 131071 } } |
| }, |
| "outputs": { |
| "values": { "dtype": "int32", "shape": [1, 100], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 100], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "dsa_index_topk_2048_decode_16k", |
| "requires": { "limits": { "maxComputeWorkgroupStorageSize": 32768 } }, |
| "provenance": { |
| "notes": "Selecting k=2048 from 16,384 candidates requires merge chunks larger than k; a chunk no larger than k does not reduce the candidate count. A 4,096-element chunk halves the candidates per stage and requires 32 KiB of workgroup storage." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2048 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [1, 16384], |
| "data": { "kind": "fillFloat32", "sinStep": 0.0131, "cosStep": 0.0237, "scale": 4.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 2048], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 2048], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "dsa_index_topk_2048_prefill_rows", |
| "requires": { "limits": { "maxComputeWorkgroupStorageSize": 32768 } }, |
| "provenance": { |
| "notes": "Several query rows each retain 2,048 of 4,096 candidate keys; the expected top-k values and indices must be correct across every row." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 2048 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [8, 4096], |
| "data": { "kind": "fillFloat32", "sinStep": 0.0131, "cosStep": 0.0237, "scale": 4.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [8, 2048], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [8, 2048], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "small_rows_batched_4096x32_k4_f16", |
| "provenance": { |
| "notes": "A float16 tensor with 4,096 rows, axis length 32, and k=4 uses a 33-value cycle of exact multiples of 1/64, leaving unambiguous top-four indices in every row." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [4096, 32], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/topk_small_rows_rank3_cycle33" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [4096, 4], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [4096, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "portable_rows_smallk_1025x128_k4_f16", |
| "provenance": { |
| "notes": "A float16 tensor with 1,025 rows, axis length 128, and k=4 exercises the portable small-k row specialization. A 129-value cycle rotates distinct, exactly representable values through each row." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [1025, 128], |
| "data": { |
| "kind": "cycle", |
| "values": [0.00390625, 0.0078125, 0.01171875, 0.015625, 0.01953125, 0.0234375, 0.02734375, 0.03125, 0.03515625, 0.0390625, 0.04296875, 0.046875, 0.05078125, 0.0546875, 0.05859375, 0.0625, 0.06640625, 0.0703125, 0.07421875, 0.078125, 0.08203125, 0.0859375, 0.08984375, 0.09375, 0.09765625, 0.1015625, 0.10546875, 0.109375, 0.11328125, 0.1171875, 0.12109375, 0.125, 0.12890625, 0.1328125, 0.13671875, 0.140625, 0.14453125, 0.1484375, 0.15234375, 0.15625, 0.16015625, 0.1640625, 0.16796875, 0.171875, 0.17578125, 0.1796875, 0.18359375, 0.1875, 0.19140625, 0.1953125, 0.19921875, 0.203125, 0.20703125, 0.2109375, 0.21484375, 0.21875, 0.22265625, 0.2265625, 0.23046875, 0.234375, 0.23828125, 0.2421875, 0.24609375, 0.25, 0.25390625, 0.2578125, 0.26171875, 0.265625, 0.26953125, 0.2734375, 0.27734375, 0.28125, 0.28515625, 0.2890625, 0.29296875, 0.296875, 0.30078125, 0.3046875, 0.30859375, 0.3125, 0.31640625, 0.3203125, 0.32421875, 0.328125, 0.33203125, 0.3359375, 0.33984375, 0.34375, 0.34765625, 0.3515625, 0.35546875, 0.359375, 0.36328125, 0.3671875, 0.37109375, 0.375, 0.37890625, 0.3828125, 0.38671875, 0.390625, 0.39453125, 0.3984375, 0.40234375, 0.40625, 0.41015625, 0.4140625, 0.41796875, 0.421875, 0.42578125, 0.4296875, 0.43359375, 0.4375, 0.44140625, 0.4453125, 0.44921875, 0.453125, 0.45703125, 0.4609375, 0.46484375, 0.46875, 0.47265625, 0.4765625, 0.48046875, 0.484375, 0.48828125, 0.4921875, 0.49609375, 0.5, 0.50390625] |
| } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [1025, 4], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1025, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "unsorted_small_rows_4096x8_k2_batched_route", |
| "provenance": { |
| "notes": "Exactly 4,096 eight-element rows with k=2 and unspecified output order use a nine-value cycle that leaves the top-two set unambiguous." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 0 }, |
| "args": { "k": 2 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4096, 8], |
| "data": { "kind": "cycle", "values": [0.1, 0.5, 0.9, 0.3, 0.7, 0.2, 0.6, 0.4, 0.8] } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [4096, 2], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [4096, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "small_rows_batched_rank3_4096rows_32axis_k4", |
| "provenance": { |
| "notes": "A rank-three tensor flattens to 4,096 rows of 32 values with k=4. A 33-value cycle leaves distinct top-four indices in every row." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [128, 32, 32], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/topk_small_rows_rank3_cycle33" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [128, 32, 4], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [128, 32, 4] } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_float32_axis256_k17_largest0", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 0 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 256], |
| "data": { "kind": "linspace", "start": 16777200.0, "end": 16777600.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_float32_axis256_k17_largest1", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 1 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 256], |
| "data": { "kind": "linspace", "start": 16777200.0, "end": 16777600.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_float32_axis257_k17_largest0", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 0 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 257], |
| "data": { "kind": "linspace", "start": 16777200.0, "end": 16777600.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_float32_axis257_k17_largest1", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 1 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [2, 257], |
| "data": { "kind": "linspace", "start": 16777200.0, "end": 16777600.0 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_float16_axis256_k17_largest0", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 0 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { "dtype": "float16", "shape": [2, 256], "data": { "kind": "linspace", "start": 1000.0, "end": 1200.0 } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_float16_axis256_k17_largest1", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 1 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { "dtype": "float16", "shape": [2, 256], "data": { "kind": "linspace", "start": 1000.0, "end": 1200.0 } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_float16_axis257_k17_largest0", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 0 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { "dtype": "float16", "shape": [2, 257], "data": { "kind": "linspace", "start": 1000.0, "end": 1200.0 } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_float16_axis257_k17_largest1", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 1 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { "dtype": "float16", "shape": [2, 257], "data": { "kind": "linspace", "start": 1000.0, "end": 1200.0 } } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_int32_axis256_k17_largest0", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 0 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 256], |
| "data": { "kind": "linspace", "start": -16777473, "end": -16777216 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "int32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_int32_axis256_k17_largest1", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 1 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 256], |
| "data": { "kind": "linspace", "start": -16777473, "end": -16777216 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "int32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_int32_axis257_k17_largest0", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 0 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 257], |
| "data": { "kind": "linspace", "start": -16777473, "end": -16777216 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "int32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_int32_axis257_k17_largest1", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 1 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "int32", |
| "shape": [2, 257], |
| "data": { "kind": "linspace", "start": -16777473, "end": -16777216 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "int32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_uint32_axis256_k17_largest0", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 0 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "uint32", |
| "shape": [2, 256], |
| "data": { "kind": "linspace", "start": 4294967040, "end": 4294967295 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_uint32_axis256_k17_largest1", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 1 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "uint32", |
| "shape": [2, 256], |
| "data": { "kind": "linspace", "start": 4294967040, "end": 4294967295 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_uint32_axis257_k17_largest0", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 0 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "uint32", |
| "shape": [2, 257], |
| "data": { "kind": "linspace", "start": 4294967040, "end": 4294967295 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "bitonic_prefix_boundary_uint32_axis257_k17_largest1", |
| "provenance": { |
| "notes": "Covers both sides of the default workgroup-width pruning boundary and a non-power-of-two retained prefix. Rounded adjacent ties retain stable index ordering while selected values vary across positions; integer cases preserve exact ordering beyond float32 integer precision." |
| }, |
| "attrs": { "axis": -1, "largest": 1 }, |
| "args": { "k": 17 }, |
| "inputs": { |
| "x": { |
| "dtype": "uint32", |
| "shape": [2, 257], |
| "data": { "kind": "linspace", "start": 4294967040, "end": 4294967295 } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float32_n257_axis0_largest0", |
| "args": { "k": 17 }, |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [257, 3], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [17, 3], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [17, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float32_n257_axis0_largest1", |
| "args": { "k": 17 }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [257, 3], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [17, 3], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [17, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float32_n257_axis1_largest0", |
| "args": { "k": 17 }, |
| "attrs": { "axis": 1, "largest": 0, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 257], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float32_n257_axis1_largest1", |
| "args": { "k": 17 }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 257], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float32_n4099_axis0_largest0", |
| "args": { "k": 70 }, |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4099, 3], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [70, 3], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [70, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float32_n4099_axis0_largest1", |
| "args": { "k": 70 }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [4099, 3], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [70, 3], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [70, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float32_n4099_axis1_largest0", |
| "args": { "k": 70 }, |
| "attrs": { "axis": 1, "largest": 0, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 4099], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 70], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 70], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float32_n4099_axis1_largest1", |
| "args": { "k": 70 }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float32", |
| "shape": [3, 4099], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [3, 70], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 70], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float16_n257_axis0_largest0", |
| "args": { "k": 17 }, |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [257, 3], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [17, 3], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [17, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float16_n257_axis0_largest1", |
| "args": { "k": 17 }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [257, 3], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [17, 3], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [17, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float16_n257_axis1_largest0", |
| "args": { "k": 17 }, |
| "attrs": { "axis": 1, "largest": 0, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [3, 257], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [3, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float16_n257_axis1_largest1", |
| "args": { "k": 17 }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [3, 257], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [3, 17], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 17], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float16_n4099_axis0_largest0", |
| "args": { "k": 70 }, |
| "attrs": { "axis": 0, "largest": 0, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [4099, 3], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [70, 3], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [70, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float16_n4099_axis0_largest1", |
| "args": { "k": 70 }, |
| "attrs": { "axis": 0, "largest": 1, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [4099, 3], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [70, 3], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [70, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float16_n4099_axis1_largest0", |
| "args": { "k": 70 }, |
| "attrs": { "axis": 1, "largest": 0, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [3, 4099], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [3, 70], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 70], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "order_keys_float16_n4099_axis1_largest1", |
| "args": { "k": 70 }, |
| "attrs": { "axis": 1, "largest": 1, "sorted": 1 }, |
| "inputs": { |
| "x": { |
| "dtype": "float16", |
| "shape": [3, 4099], |
| "data": { "kind": "cycle", "values": { "$ref": "#/fixtureArrays/order_key_extremes" } } |
| } |
| }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [3, 70], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [3, 70], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_signed_zero_all_ties_large_axis_k8", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.StableHybridSignedZero", |
| "notes": "Upstream (StableHybridSignedZero) fills the first 2k lanes of an 8192-wide row with alternating -0.0/+0.0 and asserts the top-k is indices 0..k-1 with the sign bits preserved. Alternating signed zeros across the WHOLE row is the stronger form: IEEE regards -0.0 and +0.0 as equal, so every lane ties and the first k indices must win. The ordering key must treat both signed zeros as equal while preserving the selected sign bits." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 8 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [1, 8192], "data": { "kind": "cycle", "values": [-0.0, 0.0] } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 8], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 8], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_signed_zero_all_ties_smallest_k16", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.StableSmallKSignedZero", |
| "notes": "largest=0 form of the signed-zero tie contract (upstream RunStableSignedZeroTopKCase(5000, 16, 0))." |
| }, |
| "attrs": { "axis": -1, "largest": 0, "sorted": 1 }, |
| "args": { "k": 16 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [1, 5000], "data": { "kind": "cycle", "values": [-0.0, 0.0] } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 16], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [1, 16], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_signed_zero_ties_small_rows_batched", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.StableHybridSignedZero", |
| "notes": "A compact signed-zero tie case checks the top-k values and indices within a short row." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 4 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [2, 32], "data": { "kind": "cycle", "values": [-0.0, 0.0] } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [2, 4], "tolerance": 0 }, |
| "indices": { "dtype": "uint32", "shape": [2, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_all_ties_500000_k256", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.StableHybridLargeLastAxis", |
| "notes": "Upstream StableHybridLargeLastAxis sweeps (dimension, k) pairs of all-equal values up to {500000, 256} and requires indices 0..k-1. The widest all-ties row pinned here is 43009 at k=100." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 256 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [1, 500000], "data": { "kind": "constant", "value": 1.0 } } }, |
| "outputs": { |
| "values": { "dtype": "float32", "shape": [1, 256], "tolerance": 0.000001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 256], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_f16_all_ties_248320_k65", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/math/topk_op_test.cc", |
| "test": "TopKOperator.StableHybridHalfLargeLastAxis", |
| "notes": "f16 form of StableHybridHalfLargeLastAxis ({248320, 65}); all values equal so indices 0..64 are required." |
| }, |
| "attrs": { "axis": -1, "largest": 1, "sorted": 1 }, |
| "args": { "k": 65 }, |
| "inputs": { "x": { "dtype": "float16", "shape": [1, 248320], "data": { "kind": "constant", "value": 1.0 } } }, |
| "outputs": { |
| "values": { "dtype": "float16", "shape": [1, 65], "tolerance": 0.001 }, |
| "indices": { "dtype": "uint32", "shape": [1, 65], "tolerance": 0 } |
| } |
| } |
| ] |
| } |
|
|
