diff --git a/sqlite/graphsample_insert.py b/sqlite/graphsample_insert.py index e430db05e..a5f623cba 100755 --- a/sqlite/graphsample_insert.py +++ b/sqlite/graphsample_insert.py @@ -69,7 +69,7 @@ def insert_subgraph_source( session = get_session(db_path) try: parent_relative_path = get_parent_relative_path(relative_model_path) - if sample_type == "fusible_graph": + if sample_type == "fusible_graph" or sample_type == "typical_graph": parent_parts = parent_relative_path.split("/") parent_parts = parent_parts[1:] parent_relative_path = "/".join(parent_parts) @@ -469,7 +469,7 @@ def main(args): relative_model_path=args.relative_model_path, db_path=args.db_path, ) - if args.sample_type in ["fusible_graph"]: + if args.sample_type in ["fusible_graph", "typical_graph"]: insert_dimension_generalization_source( subgraph_source_data["subgraph_uuid"], subgraph_source_data["full_graph_uuid"], diff --git a/sqlite/graphsample_insert.sh b/sqlite/graphsample_insert.sh index c8c4f1940..043ee63eb 100755 --- a/sqlite/graphsample_insert.sh +++ b/sqlite/graphsample_insert.sh @@ -7,7 +7,7 @@ TORCH_MODEL_LIST="graph_net/config/torch_samples_list.txt" PADDLE_MODEL_LIST="graph_net/config/small10_paddle_samples_list.txt" TYPICAL_GRAPH_SAMPLES_LIST="subgraph_dataset_20260203/deduplicated_subgraph_sample_list.txt" FUSIBLE_GRAPH_SAMPLES_LIST="subgraph_dataset_20260203/deduplicated_dimension_generalized_subgraph_sample_list.txt" -SOLE_OP_GRAPH_SAMPLES_LIST="subgraph_dataset_20260203/sole/solo_sample_list.txt" +SOLE_OP_GRAPH_SAMPLES_LIST="subgraph_dataset_20260203/solo_sample_list.txt" ORDER_VALUE=0 if [ ! -f "$DB_PATH" ]; then diff --git a/sqlite/migrates/create_main_tables_2026-02-02-031353.sql b/sqlite/migrates/create_main_tables_2026-02-02-031353.sql index 23d135125..de6c08507 100755 --- a/sqlite/migrates/create_main_tables_2026-02-02-031353.sql +++ b/sqlite/migrates/create_main_tables_2026-02-02-031353.sql @@ -29,7 +29,7 @@ CREATE TABLE IF NOT EXISTS graph_sample ( CREATE INDEX IF NOT EXISTS idx_relative_model_path ON graph_sample (relative_model_path); CREATE INDEX IF NOT EXISTS idx_graph_hash ON graph_sample (graph_hash); CREATE INDEX IF NOT EXISTS idx_order_value ON graph_sample (order_value); -CREATE UNIQUE INDEX IF NOT EXISTS uq_relative_model_path_repo_uid ON graph_sample (relative_model_path, repo_uid); +CREATE UNIQUE INDEX IF NOT EXISTS uq_relative_model_path_repo_uid ON graph_sample (relative_model_path, repo_uid,sample_type); -- create subgraph_source table CREATE TABLE IF NOT EXISTS subgraph_source (