Added examples for segments printing. (8c2f7758) · Commits · TNL / tnl-dev

Documentation/Examples/Algorithms/Segments/SegmentsPrintingExample-1.cpp

0 → 100644

+49 −0

Original line number	Diff line number	Diff line
		#include <iostream>
		#include <functional>
		#include <TNL/Containers/Vector.h>
		#include <TNL/Algorithms/Segments/CSR.h>
		#include <TNL/Algorithms/Segments/Ellpack.h>
		#include <TNL/Algorithms/Segments/ChunkedEllpack.h>
		#include <TNL/Algorithms/Segments/BiEllpack.h>
		#include <TNL/Devices/Host.h>
		#include <TNL/Devices/Cuda.h>

		template< typename Segments >
		void SegmentsExample()
		{
		/***
		* Create segments with given segments sizes and print their setup.
		*/
		Segments segments{ 1, 2, 3, 4, 5 };
		std::cout << "Segments sizes are: " << segments << std::endl << std::endl;
		}

		int main( int argc, char* argv[] )
		{
		std::cout << "Example of CSR segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::CSR< TNL::Devices::Host, int > >();

		std::cout << "Example of Ellpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::Ellpack< TNL::Devices::Host, int > >();

		std::cout << "Example of ChunkedEllpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::ChunkedEllpack< TNL::Devices::Host, int > >();

		std::cout << "Example of BiEllpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::BiEllpack< TNL::Devices::Host, int > >();

		#ifdef HAVE_CUDA
		std::cout << "Example of CSR segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::CSR< TNL::Devices::Cuda, int > >();

		std::cout << "Example of Ellpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::Ellpack< TNL::Devices::Cuda, int > >();

		std::cout << "Example of ChunkedEllpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::ChunkedEllpack< TNL::Devices::Cuda, int > >();

		std::cout << "Example of BiEllpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::BiEllpack< TNL::Devices::Cuda, int > >();
		#endif
		return EXIT_SUCCESS;
		}

Documentation/Examples/Algorithms/Segments/SegmentsPrintingExample-1.cu

0 → 120000

+1 −0

Original line number	Diff line number	Diff line
		SegmentsPrintingExample-1.cpp
		No newline at end of file

Documentation/Examples/Algorithms/Segments/SegmentsPrintingExample-2.cpp

0 → 100644

+68 −0

Original line number	Diff line number	Diff line
		#include <iostream>
		#include <functional>
		#include <TNL/Containers/Vector.h>
		#include <TNL/Algorithms/Segments/CSR.h>
		#include <TNL/Algorithms/Segments/Ellpack.h>
		#include <TNL/Algorithms/Segments/ChunkedEllpack.h>
		#include <TNL/Algorithms/Segments/BiEllpack.h>
		#include <TNL/Devices/Host.h>
		#include <TNL/Devices/Cuda.h>

		template< typename Segments >
		void SegmentsExample()
		{
		using Device = typename Segments::DeviceType;

		/***
		* Create segments with given segments sizes.
		*/
		TNL::Containers::Vector< int, Device > sizes{ 1, 2, 3, 4, 5 };
		Segments segments( sizes );
		std::cout << "Segments sizes are: " << segments << std::endl;

		/***
		* Allocate array for the segments;
		*/
		TNL::Containers::Array< double, Device > data( segments.getStorageSize(), 0.0 );
		data.forAllElements( [=] __cuda_callable__ ( int idx, double& value ) {
		value = idx + 1.0;
		} );

		/***
		* Print the data managed by the segments.
		*/
		auto data_view = data.getView();
		auto fetch = [=] __cuda_callable__ ( int globalIdx ) -> double { return data_view[ globalIdx ]; };
		printSegments( segments, fetch, std::cout );
		std::cout << std::endl;
		}

		int main( int argc, char* argv[] )
		{
		std::cout << "Example of CSR segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::CSR< TNL::Devices::Host, int > >();

		std::cout << "Example of Ellpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::Ellpack< TNL::Devices::Host, int > >();

		std::cout << "Example of ChunkedEllpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::ChunkedEllpack< TNL::Devices::Host, int > >();

		std::cout << "Example of BiEllpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::BiEllpack< TNL::Devices::Host, int > >();

		#ifdef HAVE_CUDA
		std::cout << "Example of CSR segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::CSR< TNL::Devices::Cuda, int > >();

		std::cout << "Example of Ellpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::Ellpack< TNL::Devices::Cuda, int > >();

		std::cout << "Example of ChunkedEllpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::ChunkedEllpack< TNL::Devices::Cuda, int > >();

		std::cout << "Example of BiEllpack segments on host: " << std::endl;
		SegmentsExample< TNL::Algorithms::Segments::BiEllpack< TNL::Devices::Cuda, int > >();
		#endif
		return EXIT_SUCCESS;
		}

Documentation/Examples/Algorithms/Segments/SegmentsPrintingExample-2.cu

0 → 120000

+1 −0

Original line number	Diff line number	Diff line
		SegmentsPrintingExample-2.cpp
		No newline at end of file