def test_sync_trans_read_tensors():
debugger_backend = d.DbgServices(
dump_file_path="../data/dump/gpu_dumps/sync_trans_true/alexnet")
_ = debugger_backend.initialize(net_name="Network Name goes here!",
is_sync_mode=True)
# parameter
info1 = d.TensorInfo(
node_name=
"Default/network-WithLossCell/_backbone-AlexNet/conv2-Conv2d/conv2.bias",
slot=0,
iteration=2,
device_id=0,
root_graph_id=0,
is_parameter=True)
# output tensor with zero slot
info2 = d.TensorInfo(
node_name=
"Default/network-WithLossCell/_backbone-AlexNet/conv3-Conv2d/Conv2D-op308",
slot=0,
iteration=2,
device_id=0,
root_graph_id=0,
is_parameter=False)
# output tensor with non-zero slot
info3 = d.TensorInfo(
node_name="Default/network-WithLossCell/_backbone-AlexNet/ReLUV2-op300",
slot=1,
iteration=2,
device_id=0,
root_graph_id=0,
is_parameter=False)
tensor_info = [info1, info2, info3]
tensor_data = debugger_backend.read_tensors(tensor_info)
print_read_tensors(tensor_info, tensor_data)
assert compare_actual_with_expected(test_name)
def main():
debugger_backend = d.DbgServices(
dump_file_path=
"/opt/nvme2n1/j00455527/dumps/sync_trans_false/032421/alexnet")
_ = debugger_backend.initialize(net_name="Network Name goes here!",
is_sync_mode=True)
# parameter
info1 = d.TensorInfo(
node_name=
"Default/network-WithLossCell/_backbone-AlexNet/conv2-Conv2d/conv2.bias",
slot=0,
iteration=2,
device_id=0,
root_graph_id=0,
is_parameter=True)
# output tensor with zero slot
info2 = d.TensorInfo(
node_name=
"Default/network-WithLossCell/_backbone-AlexNet/conv3-Conv2d/Conv2D-op168",
slot=0,
iteration=2,
device_id=0,
root_graph_id=0,
is_parameter=False)
# output tensor with non-zero slot
info3 = d.TensorInfo(
node_name="Default/network-WithLossCell/_backbone-AlexNet/ReLUV2-op346",
slot=1,
iteration=2,
device_id=0,
root_graph_id=0,
is_parameter=False)
tensor_info = [info1, info2, info3]
tensor_data = debugger_backend.read_tensors(tensor_info)
print_read_tensors(tensor_info, tensor_data)
def main():
debugger_backend = d.DbgServices(
dump_file_path="/opt/nvme2n1/j00455527/dumps/async_sink_true/032421")
_ = debugger_backend.initialize(net_name="alexnet", is_sync_mode=False)
# output tensor with zero slot
info1 = d.TensorInfo(node_name="Default/network-TrainOneStepCell/network-WithLossCell/_backbone-AlexNet/"
"conv3-Conv2d/Conv2D-op169",
slot=0, iteration=2, device_id=0, root_graph_id=1, is_parameter=False)
# output tensor with non-zero slot
info2 = d.TensorInfo(node_name="Default/network-TrainOneStepCell/network-WithLossCell/_backbone-AlexNet/"
"ReLUV2-op348",
slot=1, iteration=2, device_id=0, root_graph_id=1, is_parameter=False)
tensor_info = [info1, info2]
tensor_data = debugger_backend.read_tensors(tensor_info)
print_read_tensors(tensor_info, tensor_data)