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)