colin
/
witnn
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
witnn/Amp/trace.json

607 lines
37 KiB
JSON
Raw Permalink Blame History

{
"schemaVersion": 1,
"deviceProperties": [
{
"id": 0, "name": "NVIDIA GeForce RTX 2070 SUPER", "totalGlobalMem": 8367439872,
"computeMajor": 7, "computeMinor": 5,
"maxThreadsPerBlock": 1024, "maxThreadsPerMultiprocessor": 1024,
"regsPerBlock": 65536, "regsPerMultiprocessor": 65536, "warpSize": 32,
"sharedMemPerBlock": 49152, "sharedMemPerMultiprocessor": 65536,
"numSms": 40, "sharedMemPerBlockOptin": 65536
}
],
"traceEvents": [
{
"ph": "X", "cat": "cpu_op",
"name": "aten::empty", "pid": 1673, "tid": 1673,
"ts": 1641986274562095, "dur": 8,
"args": {
"External id": 2,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/autograd/profiler.py(432): __init__;test.py(112): <module>;", "Input type": ["GenericList", "Int", "", "", "Bool", ""]
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::zero_", "pid": 1673, "tid": 1673,
"ts": 1641986274562123, "dur": 1,
"args": {
"External id": 3,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[1]], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/autograd/profiler.py(432): __init__;test.py(112): <module>;", "Input type": ["float"]
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::zeros", "pid": 1673, "tid": 1673,
"ts": 1641986274561890, "dur": 238,
"args": {
"External id": 1,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/autograd/profiler.py(432): __init__;test.py(112): <module>;", "Input type": ["GenericList", "Int", "", "", "Bool"]
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::empty", "pid": 1673, "tid": 1673,
"ts": 1641986274562166, "dur": 1,
"args": {
"External id": 5,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/autograd/profiler.py(435): __enter__;test.py(112): <module>;", "Input type": ["GenericList", "Int", "", "", "", ""]
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::empty", "pid": 1673, "tid": 1673,
"ts": 1641986274568374, "dur": 8,
"args": {
"External id": 10,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["GenericList", "Int", "Int", "", "", ""]
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::as_strided_", "pid": 1673, "tid": 1673,
"ts": 1641986274568866, "dur": 5,
"args": {
"External id": 11,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 24, 24], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "GenericList", "GenericList", ""]
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::mkldnn_convolution", "pid": 1673, "tid": 1673,
"ts": 1641986274562855, "dur": 6026,
"args": {
"External id": 9,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 1, 28, 28], [8, 1, 5, 5], [8], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Int"], "Fwd thread id": 0, "Sequence number": 0
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::_convolution", "pid": 1673, "tid": 1673,
"ts": 1641986274562686, "dur": 6199,
"args": {
"External id": 8,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 1, 28, 28], [8, 1, 5, 5], [8], [], [], [], [], [], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Bool", "GenericList", "Int", "Bool", "Bool", "Bool", "Bool"], "Fwd thread id": 0, "Sequence number": 0
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::convolution", "pid": 1673, "tid": 1673,
"ts": 1641986274562654, "dur": 6233,
"args": {
"External id": 7,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 1, 28, 28], [8, 1, 5, 5], [8], [], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Bool", "GenericList", "Int"], "Fwd thread id": 0, "Sequence number": 0
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::conv2d", "pid": 1673, "tid": 1673,
"ts": 1641986274562628, "dur": 6264,
"args": {
"External id": 6,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 1, 28, 28], [8, 1, 5, 5], [8], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Int"], "Fwd thread id": 0, "Sequence number": 0
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::max_pool2d_with_indices", "pid": 1673, "tid": 1673,
"ts": 1641986274569358, "dur": 613,
"args": {
"External id": 13,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 24, 24], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/functional.py(719): _max_pool2d;/home/colin/anaconda3/lib/python3.8/site-packages/torch/_jit_internal.py(422): fn;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/pooling.py(162): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "GenericList", "GenericList", "GenericList", "GenericList", "Bool"], "Fwd thread id": 0, "Sequence number": 1
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::max_pool2d", "pid": 1673, "tid": 1673,
"ts": 1641986274569259, "dur": 716,
"args": {
"External id": 12,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 24, 24], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/functional.py(719): _max_pool2d;/home/colin/anaconda3/lib/python3.8/site-packages/torch/_jit_internal.py(422): fn;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/pooling.py(162): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "GenericList", "GenericList", "GenericList", "GenericList", "Bool"], "Fwd thread id": 0, "Sequence number": 1
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::sigmoid", "pid": 1673, "tid": 1673,
"ts": 1641986274570156, "dur": 301,
"args": {
"External id": 14,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 12, 12]], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/activation.py(291): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float"], "Fwd thread id": 0, "Sequence number": 2
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::empty", "pid": 1673, "tid": 1673,
"ts": 1641986274570899, "dur": 4,
"args": {
"External id": 19,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["GenericList", "Int", "Int", "", "", ""]
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::as_strided_", "pid": 1673, "tid": 1673,
"ts": 1641986274570972, "dur": 1,
"args": {
"External id": 20,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 10, 10], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "GenericList", "GenericList", ""]
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::mkldnn_convolution", "pid": 1673, "tid": 1673,
"ts": 1641986274570551, "dur": 429,
"args": {
"External id": 18,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 12, 12], [8, 8, 3, 3], [8], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Int"], "Fwd thread id": 0, "Sequence number": 3
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::_convolution", "pid": 1673, "tid": 1673,
"ts": 1641986274570531, "dur": 452,
"args": {
"External id": 17,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 12, 12], [8, 8, 3, 3], [8], [], [], [], [], [], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Bool", "GenericList", "Int", "Bool", "Bool", "Bool", "Bool"], "Fwd thread id": 0, "Sequence number": 3
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::convolution", "pid": 1673, "tid": 1673,
"ts": 1641986274570514, "dur": 476,
"args": {
"External id": 16,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 12, 12], [8, 8, 3, 3], [8], [], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Bool", "GenericList", "Int"], "Fwd thread id": 0, "Sequence number": 3
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::conv2d", "pid": 1673, "tid": 1673,
"ts": 1641986274570497, "dur": 495,
"args": {
"External id": 15,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 12, 12], [8, 8, 3, 3], [8], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Int"], "Fwd thread id": 0, "Sequence number": 3
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::max_pool2d_with_indices", "pid": 1673, "tid": 1673,
"ts": 1641986274571045, "dur": 22,
"args": {
"External id": 22,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 10, 10], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/functional.py(719): _max_pool2d;/home/colin/anaconda3/lib/python3.8/site-packages/torch/_jit_internal.py(422): fn;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/pooling.py(162): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "GenericList", "GenericList", "GenericList", "GenericList", "Bool"], "Fwd thread id": 0, "Sequence number": 4
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::max_pool2d", "pid": 1673, "tid": 1673,
"ts": 1641986274571027, "dur": 42,
"args": {
"External id": 21,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 10, 10], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/functional.py(719): _max_pool2d;/home/colin/anaconda3/lib/python3.8/site-packages/torch/_jit_internal.py(422): fn;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/pooling.py(162): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "GenericList", "GenericList", "GenericList", "GenericList", "Bool"], "Fwd thread id": 0, "Sequence number": 4
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::sigmoid", "pid": 1673, "tid": 1673,
"ts": 1641986274571090, "dur": 16,
"args": {
"External id": 23,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 5, 5]], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/activation.py(291): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float"], "Fwd thread id": 0, "Sequence number": 5
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::empty", "pid": 1673, "tid": 1673,
"ts": 1641986274571929, "dur": 4,
"args": {
"External id": 28,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["GenericList", "Int", "Int", "", "", ""]
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::as_strided_", "pid": 1673, "tid": 1673,
"ts": 1641986274573076, "dur": 1,
"args": {
"External id": 29,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 10, 1, 1], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "GenericList", "GenericList", ""]
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::mkldnn_convolution", "pid": 1673, "tid": 1673,
"ts": 1641986274571201, "dur": 1884,
"args": {
"External id": 27,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 5, 5], [10, 8, 5, 5], [10], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Int"], "Fwd thread id": 0, "Sequence number": 6
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::_convolution", "pid": 1673, "tid": 1673,
"ts": 1641986274571182, "dur": 1906,
"args": {
"External id": 26,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 5, 5], [10, 8, 5, 5], [10], [], [], [], [], [], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Bool", "GenericList", "Int", "Bool", "Bool", "Bool", "Bool"], "Fwd thread id": 0, "Sequence number": 6
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::convolution", "pid": 1673, "tid": 1673,
"ts": 1641986274571163, "dur": 1927,
"args": {
"External id": 25,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 5, 5], [10, 8, 5, 5], [10], [], [], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Bool", "GenericList", "Int"], "Fwd thread id": 0, "Sequence number": 6
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::conv2d", "pid": 1673, "tid": 1673,
"ts": 1641986274571146, "dur": 1946,
"args": {
"External id": 24,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 8, 5, 5], [10, 8, 5, 5], [10], [], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(442): _conv_forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/conv.py(446): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/container.py(141): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(95): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "float", "float", "GenericList", "GenericList", "GenericList", "Int"], "Fwd thread id": 0, "Sequence number": 6
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::view", "pid": 1673, "tid": 1673,
"ts": 1641986274573139, "dur": 17,
"args": {
"External id": 30,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 10, 1, 1], []], "Call stack": "test.py(96): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "GenericList"], "Fwd thread id": 0, "Sequence number": 7
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::_log_softmax", "pid": 1673, "tid": 1673,
"ts": 1641986274573506, "dur": 279,
"args": {
"External id": 32,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 10], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/functional.py(1769): log_softmax;test.py(98): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "Int", "Bool"], "Fwd thread id": 0, "Sequence number": 8
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::log_softmax", "pid": 1673, "tid": 1673,
"ts": 1641986274573487, "dur": 301,
"args": {
"External id": 31,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 10], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/functional.py(1769): log_softmax;test.py(98): forward;/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py(1102): _call_impl;test.py(114): <module>;", "Input type": ["float", "Int", ""], "Fwd thread id": 0, "Sequence number": 8
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::nll_loss_forward", "pid": 1673, "tid": 1673,
"ts": 1641986274574609, "dur": 20,
"args": {
"External id": 35,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 10], [12], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/functional.py(2532): nll_loss;test.py(115): <module>;", "Input type": ["float", "long int", "", "Int", "Int"], "Fwd thread id": 0, "Sequence number": 9
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::nll_loss", "pid": 1673, "tid": 1673,
"ts": 1641986274574475, "dur": 156,
"args": {
"External id": 34,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 10], [12], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/functional.py(2532): nll_loss;test.py(115): <module>;", "Input type": ["float", "long int", "", "Int", "Int"], "Fwd thread id": 0, "Sequence number": 9
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "aten::nll_loss_nd", "pid": 1673, "tid": 1673,
"ts": 1641986274574185, "dur": 458,
"args": {
"External id": 33,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Input Dims": [[12, 10], [12], [], [], []], "Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/nn/functional.py(2532): nll_loss;test.py(115): <module>;", "Input type": ["float", "long int", "", "Int", "Int"], "Fwd thread id": 0, "Sequence number": 9
}
},
{
"ph": "X", "cat": "cpu_op",
"name": "model_inference", "pid": 1673, "tid": 1673,
"ts": 1641986274562158, "dur": 12505,
"args": {
"External id": 4,
"Trace name": "PyTorch Profiler", "Trace iteration": 0,
"Call stack": "/home/colin/anaconda3/lib/python3.8/site-packages/torch/autograd/profiler.py(435): __enter__;test.py(112): <module>;"
}
},
{
"ph": "X", "cat": "Runtime",
"name": "cudaGetDeviceCount", "pid": 1673, "tid": 1673,
"ts": 1641986274574776, "dur": 524,
"args": {
"cbid": 3, "correlation": 1,
"external id": 0, "external ts": 0
}
},
{
"ph": "X", "cat": "Runtime",
"name": "cudaGetDeviceProperties", "pid": 1673, "tid": 1673,
"ts": 1641986274575567, "dur": 176,
"args": {
"cbid": 4, "correlation": 4,
"external id": 0, "external ts": 0
}
},
{
"ph": "X", "cat": "Runtime",
"name": "cudaGetDeviceCount", "pid": 1673, "tid": 1673,
"ts": 1641986274575757, "dur": 0,
"args": {
"cbid": 3, "correlation": 6,
"external id": 0, "external ts": 0
}
},
{
"ph": "X", "cat": "Runtime",
"name": "cudaGetDeviceProperties", "pid": 1673, "tid": 1673,
"ts": 1641986274575759, "dur": 157,
"args": {
"cbid": 4, "correlation": 7,
"external id": 0, "external ts": 0
}
},
{
"ph": "X", "cat": "Runtime",
"name": "cudaDeviceSynchronize", "pid": 1673, "tid": 1673,
"ts": 1641986274576391, "dur": 3469523,
"args": {
"cbid": 165, "correlation": 10,
"external id": 0, "external ts": 0
}
},
{
"name": "process_name", "ph": "M", "ts": 1641986274561617, "pid": 1673, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 1641986274561617, "pid": 1673, "tid": 0,
"args": {
"labels": "CPU"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 1641986274561617, "pid": 1673, "tid": 0,
"args": {
"sort_index": 1673
}
},
{
"name": "process_name", "ph": "M", "ts": 1641986274561617, "pid": 0, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 1641986274561617, "pid": 0, "tid": 0,
"args": {
"labels": "GPU 0"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 1641986274561617, "pid": 0, "tid": 0,
"args": {
"sort_index": 16777216
}
},
{
"name": "process_name", "ph": "M", "ts": 1641986274561617, "pid": 1, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 1641986274561617, "pid": 1, "tid": 0,
"args": {
"labels": "GPU 1"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 1641986274561617, "pid": 1, "tid": 0,
"args": {
"sort_index": 16777217
}
},
{
"name": "process_name", "ph": "M", "ts": 1641986274561617, "pid": 2, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 1641986274561617, "pid": 2, "tid": 0,
"args": {
"labels": "GPU 2"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 1641986274561617, "pid": 2, "tid": 0,
"args": {
"sort_index": 16777218
}
},
{
"name": "process_name", "ph": "M", "ts": 1641986274561617, "pid": 3, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 1641986274561617, "pid": 3, "tid": 0,
"args": {
"labels": "GPU 3"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 1641986274561617, "pid": 3, "tid": 0,
"args": {
"sort_index": 16777219
}
},
{
"name": "process_name", "ph": "M", "ts": 1641986274561617, "pid": 4, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 1641986274561617, "pid": 4, "tid": 0,
"args": {
"labels": "GPU 4"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 1641986274561617, "pid": 4, "tid": 0,
"args": {
"sort_index": 16777220
}
},
{
"name": "process_name", "ph": "M", "ts": 1641986274561617, "pid": 5, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 1641986274561617, "pid": 5, "tid": 0,
"args": {
"labels": "GPU 5"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 1641986274561617, "pid": 5, "tid": 0,
"args": {
"sort_index": 16777221
}
},
{
"name": "process_name", "ph": "M", "ts": 1641986274561617, "pid": 6, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 1641986274561617, "pid": 6, "tid": 0,
"args": {
"labels": "GPU 6"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 1641986274561617, "pid": 6, "tid": 0,
"args": {
"sort_index": 16777222
}
},
{
"name": "process_name", "ph": "M", "ts": 1641986274561617, "pid": 7, "tid": 0,
"args": {
"name": "python"
}
},
{
"name": "process_labels", "ph": "M", "ts": 1641986274561617, "pid": 7, "tid": 0,
"args": {
"labels": "GPU 7"
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 1641986274561617, "pid": 7, "tid": 0,
"args": {
"sort_index": 16777223
}
},
{
"name": "thread_name", "ph": "M", "ts": 1641986274561617, "pid": 1673, "tid": 1673,
"args": {
"name": "thread 1673 (python)"
}
},
{
"name": "thread_sort_index", "ph": "M", "ts": 1641986274561617, "pid": 1673, "tid": 1673,
"args": {
"sort_index": 1673
}
},
{
"ph": "X", "cat": "Trace", "ts": 1641986274561617, "dur": 3484317,
"pid": "Spans", "tid": "PyTorch Profiler",
"name": "PyTorch Profiler (0)",
"args": {
"Op count": 0
}
},
{
"name": "process_sort_index", "ph": "M", "ts": 1641986274561617,
"pid": "Spans", "tid": 0,
"args": {
"sort_index": 536870912
}
},
{
"name": "Iteration Start: PyTorch Profiler", "ph": "i", "s": "g",
"pid": "Traces", "tid": "Trace PyTorch Profiler", "ts": 1641986274561617
},
{
"name": "Record Window End", "ph": "i", "s": "g",
"pid": "", "tid": "", "ts": 1641986278046116
}
]}
Reference in New Issue View Git Blame Copy Permalink