Select sharding functor.

StanfordLegion · Jul 31, 2024 · fba3388 · fba3388
1 parent 95c6111
commit fba3388
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/mapper.cc b/src/mapper.cc
@@ -197,6 +197,13 @@ void FuzzMapper::select_task_sources(const MapperContext ctx, const Task &task,
   random_sources(rng, input.source_instances, output.chosen_ranking);
 }
 
+void FuzzMapper::select_sharding_functor(const MapperContext ctx, const Task &task,
+                                         const SelectShardingFunctorInput &input,
+                                         SelectShardingFunctorOutput &output) {
+  // TODO: customize the sharding functor
+  output.chosen_functor = 0;
+}
+
 void FuzzMapper::map_inline(const MapperContext ctx, const InlineMapping &inline_op,
                             const MapInlineInput &input, MapInlineOutput &output) {
   RngChannel &rng = map_inline_channel;

diff --git a/src/mapper.h b/src/mapper.h
@@ -50,6 +50,9 @@ class FuzzMapper : public NullMapper {
   void select_task_sources(const MapperContext ctx, const Task &task,
                            const SelectTaskSrcInput &input,
                            SelectTaskSrcOutput &output) override;
+  void select_sharding_functor(const MapperContext ctx, const Task &task,
+                               const SelectShardingFunctorInput &input,
+                               SelectShardingFunctorOutput &output) override;
 
 public:  // Inline mapping calls
   void map_inline(const MapperContext ctx, const InlineMapping &inline_op,