From cc5e9ca2cb84f63ba0f2abd5b51727fec9807865 Mon Sep 17 00:00:00 2001 From: peiyuan Date: Wed, 25 Sep 2024 11:09:42 +0800 Subject: [PATCH] bypass sanity check --- python/pylibwholegraph/pylibwholegraph/torch/gnn_model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/pylibwholegraph/pylibwholegraph/torch/gnn_model.py b/python/pylibwholegraph/pylibwholegraph/torch/gnn_model.py index 709060c4f..fa950cdea 100644 --- a/python/pylibwholegraph/pylibwholegraph/torch/gnn_model.py +++ b/python/pylibwholegraph/pylibwholegraph/torch/gnn_model.py @@ -162,6 +162,7 @@ def create_sub_graph( ), num_src_nodes=target_gid.size(0), num_dst_nodes=target_gid_1.size(0), + node_count_check=False, ) return block elif framework_name == "cugraph":