Diffusion tamaas (archive/legacy_mpi) Browse

tamaas/tests/test_cuda.cpp
b3c9bf19b545archive/legacy_mpi

test_cuda.cpp
View Options

			#include "tamaas.hh"
			#include "grid.hh"
			#include <algorithm>

			using namespace tamaas;

			int main() {
			initialize();
			Grid<Real, 1> grid({20}, 1);
			std::iota(grid.begin(), grid.end(), 0);
			Loop::loop([] CUDA_LAMBDA (Real & x) {
			x += 1;
			}, grid);
			std::cout << grid << std::endl;
			finalize();
			return 0;
			}