Spike/sskj-h3
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
sskj-h3/sglang-profile/results/h3-sdpa-kernel-matrix-20260825-run1/jobs/gpu6-RVA-efficient-contiguous.trace.json

586 lines
20 KiB
JSON
Raw Blame History

{
"schemaVersion": 1,
"deviceProperties": [
{
"id": 0, "name": "NVIDIA RTX 6000D", "totalGlobalMem": 89171361792,
"computeMajor": 12, "computeMinor": 0,
"maxThreadsPerBlock": 1024, "maxThreadsPerMultiprocessor": 1536,
"regsPerBlock": 65536, "warpSize": 32,
"sharedMemPerBlock": 49152, "numSms": 156
, "regsPerMultiprocessor": 65536, "sharedMemPerBlockOptin": 101376, "sharedMemPerMultiprocessor": 102400
}
],
"trace_id": "19D3D8DB658043E9B75B3449AF42BD40",
"record_shapes": 1,
"cuda_driver_version": 13000,
"cuda_runtime_version": 13000,
"cupti_version": 130001,
"displayTimeUnit": "ms",
"baseTimeNanoseconds": 1782967788000000000,
"traceEvents": [
{
"ph": "X", "cat": "cpu_op", "name": "aten::scaled_dot_product_attention", "pid": 2815991, "tid": 2815991,
"ts": 4657016546297.190, "dur": 2126.626,
"args": {
"External id": 513,"Sequence number": 0, "Fwd thread id": 0, "Record function id": 0, "Concrete Inputs": ["", "", "", "", "0.", "False", "", "False"], "Input type": ["c10::BFloat16", "c10::BFloat16", "c10::BFloat16", "", "Scalar", "Scalar", "", "Scalar"], "Input Strides": [[292264448, 10438016, 128, 1], [292264448, 10438016, 128, 1], [292264448, 10438016, 128, 1], [], [], [], [], []], "Input Dims": [[1, 28, 81547, 128], [1, 28, 81547, 128], [1, 28, 81547, 128], [], [], [], [], []], "Ev Idx": 0
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::_scaled_dot_product_efficient_attention", "pid": 2815991, "tid": 2815991,
"ts": 4657016546326.000, "dur": 2095.833,
"args": {
"External id": 514,"Sequence number": 0, "Fwd thread id": 0, "Record function id": 0, "Concrete Inputs": ["", "", "", "", "False", "0.", "False", ""], "Input type": ["c10::BFloat16", "c10::BFloat16", "c10::BFloat16", "", "Scalar", "Scalar", "Scalar", ""], "Input Strides": [[292264448, 10438016, 128, 1], [292264448, 10438016, 128, 1], [292264448, 10438016, 128, 1], [], [], [], [], []], "Input Dims": [[1, 28, 81547, 128], [1, 28, 81547, 128], [1, 28, 81547, 128], [], [], [], [], []], "Ev Idx": 1
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::transpose", "pid": 2815991, "tid": 2815991,
"ts": 4657016546337.629, "dur": 24.518,
"args": {
"External id": 515,"Record function id": 0, "Concrete Inputs": ["", "1", "2"], "Input type": ["c10::BFloat16", "Scalar", "Scalar"], "Input Strides": [[292264448, 10438016, 128, 1], [], []], "Input Dims": [[1, 28, 81547, 128], [], []], "Ev Idx": 2
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::as_strided", "pid": 2815991, "tid": 2815991,
"ts": 4657016546353.196, "dur": 5.920,
"args": {
"External id": 516,"Record function id": 0, "Concrete Inputs": ["", "[1, 81547, 28, 128]", "[292264448, 128, 10438016, 1]", ""], "Input type": ["c10::BFloat16", "ScalarList", "ScalarList", ""], "Input Strides": [[292264448, 10438016, 128, 1], [], [], []], "Input Dims": [[1, 28, 81547, 128], [], [], []], "Ev Idx": 3
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::transpose", "pid": 2815991, "tid": 2815991,
"ts": 4657016546363.014, "dur": 2.698,
"args": {
"External id": 517,"Record function id": 0, "Concrete Inputs": ["", "1", "2"], "Input type": ["c10::BFloat16", "Scalar", "Scalar"], "Input Strides": [[292264448, 10438016, 128, 1], [], []], "Input Dims": [[1, 28, 81547, 128], [], []], "Ev Idx": 4
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::as_strided", "pid": 2815991, "tid": 2815991,
"ts": 4657016546365.018, "dur": 0.425,
"args": {
"External id": 518,"Record function id": 0, "Concrete Inputs": ["", "[1, 81547, 28, 128]", "[292264448, 128, 10438016, 1]", ""], "Input type": ["c10::BFloat16", "ScalarList", "ScalarList", ""], "Input Strides": [[292264448, 10438016, 128, 1], [], [], []], "Input Dims": [[1, 28, 81547, 128], [], [], []], "Ev Idx": 5
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::transpose", "pid": 2815991, "tid": 2815991,
"ts": 4657016546366.499, "dur": 1.598,
"args": {
"External id": 519,"Record function id": 0, "Concrete Inputs": ["", "1", "2"], "Input type": ["c10::BFloat16", "Scalar", "Scalar"], "Input Strides": [[292264448, 10438016, 128, 1], [], []], "Input Dims": [[1, 28, 81547, 128], [], []], "Ev Idx": 6
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::as_strided", "pid": 2815991, "tid": 2815991,
"ts": 4657016546367.481, "dur": 0.362,
"args": {
"External id": 520,"Record function id": 0, "Concrete Inputs": ["", "[1, 81547, 28, 128]", "[292264448, 128, 10438016, 1]", ""], "Input type": ["c10::BFloat16", "ScalarList", "ScalarList", ""], "Input Strides": [[292264448, 10438016, 128, 1], [], [], []], "Input Dims": [[1, 28, 81547, 128], [], [], []], "Ev Idx": 7
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::_efficient_attention_forward", "pid": 2815991, "tid": 2815991,
"ts": 4657016546373.037, "dur": 2038.377,
"args": {
"External id": 521,"Record function id": 0, "Concrete Inputs": ["", "", "", "", "", "", "", "", "0.", "0", "False", "", "", ""], "Input type": ["c10::BFloat16", "c10::BFloat16", "c10::BFloat16", "", "", "", "", "", "Scalar", "Scalar", "Scalar", "", "", ""], "Input Strides": [[292264448, 128, 10438016, 1], [292264448, 128, 10438016, 1], [292264448, 128, 10438016, 1], [], [], [], [], [], [], [], [], [], [], []], "Input Dims": [[1, 81547, 28, 128], [1, 81547, 28, 128], [1, 81547, 28, 128], [], [], [], [], [], [], [], [], [], [], []], "Ev Idx": 8
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::empty", "pid": 2815991, "tid": 2815991,
"ts": 4657016548051.351, "dur": 8.147,
"args": {
"External id": 522,"Record function id": 0, "Concrete Inputs": ["[]", "4", "", "", "", ""], "Input type": ["ScalarList", "Scalar", "", "", "", ""], "Input Strides": [[], [], [], [], [], []], "Input Dims": [[], [], [], [], [], []], "Ev Idx": 9
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::empty", "pid": 2815991, "tid": 2815991,
"ts": 4657016548061.117, "dur": 1.001,
"args": {
"External id": 523,"Record function id": 0, "Concrete Inputs": ["[]", "4", "", "", "", ""], "Input type": ["ScalarList", "Scalar", "", "", "", ""], "Input Strides": [[], [], [], [], [], []], "Input Dims": [[], [], [], [], [], []], "Ev Idx": 10
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::empty", "pid": 2815991, "tid": 2815991,
"ts": 4657016548063.978, "dur": 283.117,
"args": {
"External id": 524,"Record function id": 0, "Concrete Inputs": ["[1, 81547, 28, 128]", "15", "0", "", "", ""], "Input type": ["ScalarList", "Scalar", "Scalar", "", "", ""], "Input Strides": [[], [], [], [], [], []], "Input Dims": [[], [], [], [], [], []], "Ev Idx": 11
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::empty", "pid": 2815991, "tid": 2815991,
"ts": 4657016548349.724, "dur": 3.193,
"args": {
"External id": 525,"Record function id": 0, "Concrete Inputs": ["[1, 28, 0]", "6", "0", "", "", ""], "Input type": ["ScalarList", "Scalar", "Scalar", "", "", ""], "Input Strides": [[], [], [], [], [], []], "Input Dims": [[], [], [], [], [], []], "Ev Idx": 12
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::transpose", "pid": 2815991, "tid": 2815991,
"ts": 4657016548413.979, "dur": 5.346,
"args": {
"External id": 526,"Record function id": 0, "Concrete Inputs": ["", "1", "2"], "Input type": ["c10::BFloat16", "Scalar", "Scalar"], "Input Strides": [[292264448, 3584, 128, 1], [], []], "Input Dims": [[1, 81547, 28, 128], [], []], "Ev Idx": 13
}
},
{
"ph": "X", "cat": "cpu_op", "name": "aten::as_strided", "pid": 2815991, "tid": 2815991,
"ts": 4657016548417.451, "dur": 1.481,
"args": {
"External id": 527,"Record function id": 0, "Concrete Inputs": ["", "[1, 28, 81547, 128]", "[292264448, 128, 3584, 1]", ""], "Input type": ["c10::BFloat16", "ScalarList", "ScalarList", ""], "Input Strides": [[292264448, 3584, 128, 1], [], [], []], "Input Dims": [[1, 81547, 28, 128], [], [], []], "Ev Idx": 14
}
},
{
"ph": "X", "cat": "overhead", "name": "Activity Buffer Request", "pid": -1, "tid": 0,
"ts": 4657016546381.473, "dur": 1653.716
},
{
"ph": "X", "cat": "cuda_runtime", "name": "cudaStreamIsCapturing", "pid": 2815991, "tid": 2815991,
"ts": 4657016548038.645, "dur": 3.334,
"args": {
"External id": 521, "cbid": 317, "correlation": 1940
}
},
{
"ph": "f", "id": 1940, "pid": 2815991, "tid": 2815991, "ts": 4657016548038.645,
"cat": "ac2g", "name": "ac2g", "bp": "e"
},
{
"ph": "X", "cat": "cuda_runtime", "name": "cudaStreamIsCapturing", "pid": 2815991, "tid": 2815991,
"ts": 4657016548078.645, "dur": 0.353,
"args": {
"External id": 524, "cbid": 317, "correlation": 1946
}
},
{
"ph": "f", "id": 1946, "pid": 2815991, "tid": 2815991, "ts": 4657016548078.645,
"cat": "ac2g", "name": "ac2g", "bp": "e"
},
{
"ph": "X", "cat": "cuda_runtime", "name": "cudaMalloc", "pid": 2815991, "tid": 2815991,
"ts": 4657016548079.434, "dur": 261.515,
"args": {
"External id": 524, "cbid": 20, "correlation": 1947
}
},
{
"ph": "f", "id": 1947, "pid": 2815991, "tid": 2815991, "ts": 4657016548079.434,
"cat": "ac2g", "name": "ac2g", "bp": "e"
},
{
"ph": "X", "cat": "cuda_runtime", "name": "cudaFuncSetAttribute", "pid": 2815991, "tid": 2815991,
"ts": 4657016548355.249, "dur": 9.717,
"args": {
"External id": 521, "cbid": 273, "correlation": 1959
}
},
{
"ph": "f", "id": 1959, "pid": 2815991, "tid": 2815991, "ts": 4657016548355.249,
"cat": "ac2g", "name": "ac2g", "bp": "e"
},
{
"ph": "X", "cat": "kernel", "name": "fmha_cutlassF_bf16_aligned_64x128_rf_sm80(PyTorchMemEffAttention::AttentionKernel<cutlass::bfloat16_t, cutlass::arch::Sm80, true, 64, 128, 128, true, true>::Params)", "pid": 0, "tid": 7,
"ts": 4657016548410.446, "dur": 1347988.562,
"args": {
"External id": 521, "queued": 0, "device": 0, "context": 1, "stream": 7, "correlation": 1961, "registers per thread": 126, "shared memory": 52224, "blocks per SM": 228.846161, "warps per SM": 1830.769287, "grid": [1275, 28, 1], "block": [32, 8, 1], "est. achieved occupancy %": 0
}
},
{
"ph": "f", "id": 1961, "pid": 0, "tid": 7, "ts": 4657016548410.446,
"cat": "ac2g", "name": "ac2g", "bp": "e"
},
{
"ph": "X", "cat": "cuda_runtime", "name": "cudaLaunchKernel", "pid": 2815991, "tid": 2815991,
"ts": 4657016548370.351, "dur": 38.474,
"args": {
"External id": 521, "cbid": 211, "correlation": 1961
}
},
{
"ph": "s", "id": 1961, "pid": 2815991, "tid": 2815991, "ts": 4657016548370.351,
"cat": "ac2g", "name": "ac2g"
},
{
"ph": "X", "cat": "cuda_runtime", "name": "cudaDeviceSynchronize", "pid": 2815991, "tid": 2815991,
"ts": 4657016548502.635, "dur": 1347903.969,
"args": {
"cbid": 165, "correlation": 1971
}
},
{
"ph": "s", "id": 1971, "pid": 2815991, "tid": 2815991, "ts": 4657016548502.635,
"cat": "ac2g", "name": "ac2g"
},
{
"ph": "X", "cat": "cuda_runtime", "name": "cudaDeviceSynchronize", "pid": 2815991, "tid": 2815991,
"ts": 4657017896434.607, "dur": 4.137,
"args": {
"cbid": 165, "correlation": 1975
}
},
{
"ph": "s", "id": 1975, "pid": 2815991, "tid": 2815991, "ts": 4657017896434.607,
"cat": "ac2g", "name": "ac2g"
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 2815991, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 2815991, "tid": 0,
"args": {
"labels": "CPU"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 2815991, "tid": 0,
"args": {
"sort_index": 2815991
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 0, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 0, "tid": 0,
"args": {
"labels": "GPU 0"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 0, "tid": 0,
"args": {
"sort_index": 5000000
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 1, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 1, "tid": 0,
"args": {
"labels": "GPU 1"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 1, "tid": 0,
"args": {
"sort_index": 5000001
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 2, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 2, "tid": 0,
"args": {
"labels": "GPU 2"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 2, "tid": 0,
"args": {
"sort_index": 5000002
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 3, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 3, "tid": 0,
"args": {
"labels": "GPU 3"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 3, "tid": 0,
"args": {
"sort_index": 5000003
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 4, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 4, "tid": 0,
"args": {
"labels": "GPU 4"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 4, "tid": 0,
"args": {
"sort_index": 5000004
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 5, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 5, "tid": 0,
"args": {
"labels": "GPU 5"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 5, "tid": 0,
"args": {
"sort_index": 5000005
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 6, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 6, "tid": 0,
"args": {
"labels": "GPU 6"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 6, "tid": 0,
"args": {
"sort_index": 5000006
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 7, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 7, "tid": 0,
"args": {
"labels": "GPU 7"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 7, "tid": 0,
"args": {
"sort_index": 5000007
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 8, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 8, "tid": 0,
"args": {
"labels": "GPU 8"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 8, "tid": 0,
"args": {
"sort_index": 5000008
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 9, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 9, "tid": 0,
"args": {
"labels": "GPU 9"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 9, "tid": 0,
"args": {
"sort_index": 5000009
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 10, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 10, "tid": 0,
"args": {
"labels": "GPU 10"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 10, "tid": 0,
"args": {
"sort_index": 5000010
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 11, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 11, "tid": 0,
"args": {
"labels": "GPU 11"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 11, "tid": 0,
"args": {
"sort_index": 5000011
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 12, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 12, "tid": 0,
"args": {
"labels": "GPU 12"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 12, "tid": 0,
"args": {
"sort_index": 5000012
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 13, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 13, "tid": 0,
"args": {
"labels": "GPU 13"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 13, "tid": 0,
"args": {
"sort_index": 5000013
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 14, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 14, "tid": 0,
"args": {
"labels": "GPU 14"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 14, "tid": 0,
"args": {
"sort_index": 5000014
}
},
{
"name": "process_name", "ph": "M", "ts": 4657016545992.475, "pid": 15, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 4657016545992.475, "pid": 15, "tid": 0,
"args": {
"labels": "GPU 15"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 15, "tid": 0,
"args": {
"sort_index": 5000015
}
},
{
"name": "thread_name", "ph": "M", "ts": 4657016545992.475, "pid": 0, "tid": 7,
"args": {
"name": "stream 7 "
}
},
{
"name": "thread_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 0, "tid": 7,
"args": {
"sort_index": 7
}
},
{
"name": "thread_name", "ph": "M", "ts": 4657016545992.475, "pid": 2815991, "tid": 2815991,
"args": {
"name": "thread 2815991 (python)"
}
},
{
"name": "thread_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 2815991, "tid": 2815991,
"args": {
"sort_index": 2815991
}
},
{
"name": "thread_name", "ph": "M", "ts": 4657016545992.475, "pid": 2815991, "tid": 2815991,
"args": {
"name": "thread 2815991 (python)"
}
},
{
"name": "thread_sort_index", "ph": "M", "ts": 4657016545992.475, "pid": 2815991, "tid": 2815991,
"args": {
"sort_index": 2815991
}
},
{
"ph": "X", "cat": "Trace", "ts": 4657016545941.061, "dur": 1350506.118,
"pid": "Spans", "tid": "PyTorch Profiler",
"name": "PyTorch Profiler (0)",
"args": {
"Op count": 0
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 4657016545941.061,
"pid": "Spans", "tid": 0,
"args": {
"sort_index": 536870912
}
},
{
"name": "Iteration Start: PyTorch Profiler", "ph": "i", "s": "g",
"pid": "Traces", "tid": "Trace PyTorch Profiler", "ts": 4657016545941.061
},
{
"name": "Record Window End", "ph": "i", "s": "g",
"pid": "", "tid": "", "ts": 4657017896666.169
}
],
"traceName": "/data/wxy/profile_results/h3-sdpa-kernel-matrix-20260825-run1/jobs/gpu6-RVA-efficient-contiguous.trace.json"
}
Reference in New Issue View Git Blame Copy Permalink