Skip to content

Synthetic diffusion MGN example. #198

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
YohannDudouit wants to merge 10 commits into
base: develop
Choose a base branch
from
Open

Synthetic diffusion MGN example. #198

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
83e0e82
MGN diffusion test.
YohannDudouit May 29, 2026
d4c7047
Improve model and training.
YohannDudouit May 29, 2026
8324d33
REfine model.
YohannDudouit May 29, 2026
cb6617a
load checkpoint fix.
YohannDudouit May 29, 2026
839e268
MAke diffusion MGN test not appear during build phase.
YohannDudouit Jun 1, 2026
278167c
Add documentation.
YohannDudouit Jun 1, 2026
5990597
clang-format.
YohannDudouit Jun 8, 2026
84d7342
Add extensive documentation to diffusion MGN python script.
YohannDudouit Jun 9, 2026
dac73a0
Add extensive documentation to the C++ diffusion MGN example.
YohannDudouit Jun 9, 2026
1294d99
Fix MGN with new global_features rank 1.
YohannDudouit Jun 9, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 18 additions & 0 deletions tests/AMSlib/ams_interface/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,3 +33,21 @@ ADD_AMS_UNIT_TEST(AMS_GRAPH_FALLBACK ams_graph_fallback)
BUILD_UNIT_TEST(ams_graph_surrogate test_graph_surrogate.cpp)
ADD_AMS_UNIT_TEST(AMS_GRAPH_SURROGATE ams_graph_surrogate)

BUILD_UNIT_TEST(ams_graph_mgn_surrogate test_graph_mgn_surrogate.cpp)
target_link_libraries(ams_graph_mgn_surrogate PRIVATE nlohmann_json::nlohmann_json)
target_compile_definitions(ams_graph_mgn_surrogate
PRIVATE
AMS_MGN_DIFFUSION_FIXTURE_DIR="${AMS_TEST_ROOT}/models/mgn_graph_diffusion"
)

# Final runtime check for the MGN diffusion workflow: fixtures mode must have
# produced the TorchScript model and binary graph cases before AMS parity runs.
add_test(
NAME MGN_DIFFUSION_AMS_PARITY
COMMAND $<TARGET_FILE:ams_graph_mgn_surrogate> -s --reporter console
)
set_tests_properties(MGN_DIFFUSION_AMS_PARITY PROPERTIES
DEPENDS MGN_DIFFUSION_FIXTURES
LABELS "AMS_INTERFACE;MGN_DIFFUSION"
WORKING_DIRECTORY "${CMAKE_CURRENT_BINARY_DIR}"
)
21 changes: 2 additions & 19 deletions tests/AMSlib/ams_interface/test_graph_fallback.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -124,21 +124,9 @@ CATCH_TEST_CASE("AMSHomogeneousGraph validates construction",
AMSInit();

CATCH_REQUIRE_NOTHROW(makeValidGraph());
{
AMSHomogeneousGraph graph(makeNodeFeatures(),
makeEdgeIndex32(),
makeEdgeFeatures());
CATCH_REQUIRE(graph.global_features.shape().size() == 1);
CATCH_REQUIRE(graph.global_features.shape()[0] == 0);
}
CATCH_REQUIRE_NOTHROW(AMSHomogeneousGraph(makeNodeFeatures(),
makeEdgeIndex64(),
makeEdgeFeatures(),
makeTensor<float>({0})));
CATCH_REQUIRE_NOTHROW(AMSHomogeneousGraph(makeNodeFeatures(),
makeEdgeIndex64(),
makeEdgeFeatures(),
makeGlobalFeatures()));
makeEdgeIndex32(),
makeEdgeFeatures()));

CATCH_REQUIRE_THROWS_AS(AMSHomogeneousGraph(makeTensor<int64_t>({3, 2}),
makeEdgeIndex64(),
Expand Down Expand Up @@ -172,11 +160,6 @@ CATCH_TEST_CASE("AMSHomogeneousGraph validates construction",
makeEdgeIndex64(),
makeTensor<int64_t>({2, 1})),
std::runtime_error);
CATCH_REQUIRE_THROWS_AS(AMSHomogeneousGraph(makeNodeFeatures(),
makeEdgeIndex64(),
makeEdgeFeatures(),
makeTensor<float>({1, 2})),
std::runtime_error);
CATCH_REQUIRE_THROWS_AS(AMSHomogeneousGraph(makeNodeFeatures(),
makeEdgeIndex64(),
makeEdgeFeatures(),
Expand Down
Loading
Loading