Task 1 (Max-Pooling using Submatrix Scattering) Assignment

Scaffold Head

#include <algorithm> #include <iostream> #include <cstdint> #include <vector> #include <cmath> #include <assert.h> #include <mpi.h> int main (int argc, char * argv[]) { MPI::Init(argc, argv); const uint64_t height = 12; const uint64_t width = 6; const uint64_t num_ranks = MPI::COMM_WORLD.Get_size(); const uint64_t rank = MPI::COMM_WORLD.Get_rank(); const uint64_t radix = uint64_t(std::sqrt(num_ranks)); const uint64_t verti = rank / radix; const uint64_t horiz = rank % radix; const bool is_root = rank == 0; assert(num_ranks == radix*radix); assert(height % radix == 0); assert(width % radix == 0); const uint64_t local_height = height / radix; const uint64_t local_width = width / radix; std::vector<double> image(height*width*is_root); std::vector<double> tile (local_height*local_width); if (is_root) { std::iota(image.begin(), image.end(), 0); std::cout << "original data:" << std::endl; for (uint64_t row = 0; row < height; ++row) for (uint64_t col = 0; col < width; ++col) std::cout << (col == 0 ? "|\t" : "") << image[row*width+col] << (col+1 == width ? "\t|\n" : "\t"); std::cout << std::endl; } // create tile data type MPI::Datatype tile_t = MPI::DOUBLE.Create_vector (local_height, local_width , width) .Create_resized(0, sizeof(double)); tile_t.Commit();

Scaffold Foot

// scatter the tiles MPI::COMM_WORLD.Scatterv(image.data(), counts, displs, tile_t, tile.data(), local_height*local_width, MPI::DOUBLE, 0); // max pooling on that tile const auto mu_ptr = std::max_element(tile.begin(), tile.end()); for (auto& x : tile) x = *mu_ptr; // gather the tiles MPI::COMM_WORLD.Gatherv( tile.data(), local_height*local_width,MPI::DOUBLE, image.data(), counts, displs, tile_t, 0); if (is_root) { std::cout << "after max pooling:" << std::endl; for (uint64_t row = 0; row < height; ++row) for (uint64_t col = 0; col < width; ++col) std::cout << (col == 0 ? "|\t" : "") << image[row*width+col] << (col+1 == width ? "\t|\n" : "\t"); } MPI::Finalize(); }

Start time:

Mo 22 Okt 2018 10:51:00

End time:

Fr 01 Mär 2019 12:00:00

General test timeout:

10.0 seconds

Tests

original data:
|    0	1	2	3	4	5	|
|	6	7	8	9	10	11	|
|	12	13	14	15	16	17	|
|	18	19	20	21	22	23	|
|	24	25	26	27	28	29	|
|	30	31	32	33	34	35	|
|	36	37	38	39	40	41	|
|	42	43	44	45	46	47	|
|	48	49	50	51	52	53	|
|	54	55	56	57	58	59	|
|	60	61	62	63	64	65	|
|	66	67	68	69	70	71	|

after max pooling:
|	32	32	32	35	35	35	|
|	32	32	32	35	35	35	|
|	32	32	32	35	35	35	|
|	32	32	32	35	35	35	|
|	32	32	32	35	35	35	|
|	32	32	32	35	35	35	|
|	68	68	68	71	71	71	|
|	68	68	68	71	71	71	|
|	68	68	68	71	71	71	|
|	68	68	68	71	71	71	|
|	68	68	68	71	71	71	|
|	68	68	68	71	71	71	|

original data:
|    0	1	2	3	4	5	|
|	6	7	8	9	10	11	|
|	12	13	14	15	16	17	|
|	18	19	20	21	22	23	|
|	24	25	26	27	28	29	|
|	30	31	32	33	34	35	|
|	36	37	38	39	40	41	|
|	42	43	44	45	46	47	|
|	48	49	50	51	52	53	|
|	54	55	56	57	58	59	|
|	60	61	62	63	64	65	|
|	66	67	68	69	70	71	|

after max pooling:
|	19	19	21	21	23	23	|
|	19	19	21	21	23	23	|
|	19	19	21	21	23	23	|
|	19	19	21	21	23	23	|
|	43	43	45	45	47	47	|
|	43	43	45	45	47	47	|
|	43	43	45	45	47	47	|
|	43	43	45	45	47	47	|
|	67	67	69	69	71	71	|
|	67	67	69	69	71	71	|
|	67	67	69	69	71	71	|
|	67	67	69	69	71	71	|