This is an automated email from the ASF dual-hosted git repository.
jroesch pushed a commit to branch ci-docker-staging
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/ci-docker-staging by this push:
new ecf8b65 Format
ecf8b65 is described below
commit ecf8b656abcfb0d73903aee9a0ee3dea039c0ff8
Author: Jared Roesch <[email protected]>
AuthorDate: Wed Oct 14 13:31:27 2020 -0700
Format
---
tutorials/frontend/build_gcn.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/tutorials/frontend/build_gcn.py b/tutorials/frontend/build_gcn.py
index a99ed80..b832d18 100644
--- a/tutorials/frontend/build_gcn.py
+++ b/tutorials/frontend/build_gcn.py
@@ -242,7 +242,9 @@ import networkx as nx
def prepare_params(g, data):
params = {}
- params["infeats"] = data.features.numpy().astype("float32") # Only
support float32 as feature for now
+ params["infeats"] = data.features.numpy().astype(
+ "float32"
+ ) # Only support float32 as feature for now
# Generate adjacency matrix
adjacency = nx.to_scipy_sparse_matrix(g)
@@ -351,5 +353,6 @@ acc = evaluate(data, logits_tvm)
print("Test accuracy of TVM results: {:.2%}".format(acc))
import tvm.testing
+
# Verify the results with the DGL model
tvm.testing.assert_allclose(logits_torch, logits_tvm, atol=1e-3)