d73e36a44ab3fa760d385c881db0fa393a557a65,torch/quantization/ns/numeric_suite_core_apis_fx.py,,compare_weights,#Any#Any#Any#Any#,117

Before Change

    matched_subgraph_pairs = get_matching_subgraph_pairs(gm_a, gm_b)

    // split the subgraph pairs into one data structure for each model
    nodes_and_names_to_instrument_a: List[Tuple[Node, str]] = []
    nodes_and_names_to_instrument_b: List[Tuple[Node, str]] = []
    for match_name, match in matched_subgraph_pairs.items():
        (node_start_a, node_end_a), (node_start_b, node_end_b) = match
        nodes_and_names_to_instrument_a.append((node_start_a, match_name))

After Change

        get_type_a_related_to_b(base_name_to_sets_of_related_ops)

    tracer_a, tracer_b = NSTracer(), NSTracer()
    gm_a = GraphModule(model_a, tracer_a.trace(model_a))
    gm_b = GraphModule(model_b, tracer_b.trace(model_b))
    return _compare_weights_impl(model_name_a, gm_a, model_name_b, gm_b)

Instances