Skip to content
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

Update to LoopTree to Python interface #294

Merged
merged 13 commits into from
Jan 21, 2025
Merged

Update to LoopTree to Python interface #294

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
b6c1644
Add functions to compute equivalent ranks and tensor size
Sep 26, 2024
b56b6c1
Merge branch 'NVlabs:master' into looptree-mapper
gilbertmike Sep 26, 2024
6318fe2
Merge branch 'NVlabs:master' into looptree-mapper
gilbertmike Sep 30, 2024
a0e5f6f
Remove boost dependency from isl-functions
Sep 30, 2024
75a90c5
Add utility method to calculate operation space volume
Oct 1, 2024
5fc06e2
Merge branch 'looptree-mapper' of github.com:gilbertmike/timeloop int…
Oct 1, 2024
6fe0ec2
Merge branch 'master' of https://github.com/NVlabs/timeloop into loop…
Oct 1, 2024
e062ed4
Fix parsing bug where tile_size is looked up instead of tile_shape
Oct 8, 2024
a6846ac
Remove dependency on boost log and filesystem libraries
Oct 28, 2024
f6ddd12
Merge branch 'master' into looptree-mapper
gilbertmike Oct 28, 2024
a613be2
Merge branch 'NVlabs:master' into looptree-mapper
gilbertmike Nov 20, 2024
fb8cc29
LoopTree to Python interface update
Dec 5, 2024
bc88e2e
Merge branch 'NVlabs:master' into looptree-mapper
gilbertmike Dec 28, 2024
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
4 changes: 2 additions & 2 deletions include/applications/looptree-model/model.hpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,12 +31,12 @@ class LooptreeModel
std::map<
std::tuple<mapping::BufferId, problem::DataSpaceId, mapping::NodeID>,
std::tuple<std::vector<analysis::SpaceTime>, std::string>
> fills_by_parent;
> reads_to_parent;

std::map<
std::tuple<mapping::BufferId, problem::DataSpaceId, mapping::NodeID>,
std::tuple<std::vector<analysis::SpaceTime>, std::string>
> fills_by_peer;
> reads_to_peer;

std::map<
std::tuple<mapping::BufferId, problem::DataSpaceId, mapping::NodeID>,
Expand Down
6 changes: 3 additions & 3 deletions src/applications/looptree-model/model.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -73,14 +73,14 @@ LooptreeModel::Result LooptreeModel::Run()
isl_pw_qpolynomial_free(p_occ_count);

auto p_fill_count = isl_map_card(stats.fill.map.copy());
model_result.fills_by_parent[key] = std::make_pair(
model_result.fills[key] = std::make_pair(
stats.fill.dim_in_tags,
isl_pw_qpolynomial_to_str(p_fill_count)
);
isl_pw_qpolynomial_free(p_fill_count);

auto p_parent_reads_count = isl_map_card(stats.parent_reads.map.copy());
model_result.fills_by_parent[key] = std::make_pair(
model_result.reads_to_parent[key] = std::make_pair(
stats.parent_reads.dim_in_tags,
isl_pw_qpolynomial_to_str(p_parent_reads_count)
);
Expand All @@ -91,7 +91,7 @@ LooptreeModel::Result LooptreeModel::Run()
p_peer_fills_count,
stats.fill.map.domain().release()
);
model_result.fills_by_peer[key] = std::make_pair(
model_result.reads_to_peer[key] = std::make_pair(
stats.link_transfer.dim_in_tags,
isl_pw_qpolynomial_to_str(p_peer_fills_count)
);
Expand Down