Merge pull request #406 from stanmoore1/kokkos_bugfix (ffe02d20) · Commits · 郑智淋 / lammps

src/KOKKOS/comm_kokkos.cpp

+3 −0

Original line number	Original line	Diff line number	Diff line
	@@ -93,6 +93,9 @@ CommKokkos::~CommKokkos()

	void CommKokkos::init()		void CommKokkos::init()
	{		{
			maxsend = BUFMIN;
			maxrecv = BUFMIN;

	grow_send_kokkos(maxsend+bufextra,0,Host);		grow_send_kokkos(maxsend+bufextra,0,Host);
	grow_recv_kokkos(maxrecv,Host);		grow_recv_kokkos(maxrecv,Host);

+5 −2

Original line number	Original line	Diff line number	Diff line
	@@ -1129,6 +1129,11 @@ void Neighbor::morph_halffull()

	if (!irq->half) continue;		if (!irq->half) continue;

			// Kokkos doesn't yet support half from full

			if (irq->kokkos_host) continue;
			if (irq->kokkos_device) continue;

	// these lists are created other ways, no need for halffull		// these lists are created other ways, no need for halffull
	// do want to process skip lists		// do want to process skip lists

	@@ -1154,8 +1159,6 @@ void Neighbor::morph_halffull()
	// this includes custom cutoff set by requestor		// this includes custom cutoff set by requestor
	// no need to check respaouter b/c it stores same pairs		// no need to check respaouter b/c it stores same pairs
	// no need to check dnum b/c only set for history		// no need to check dnum b/c only set for history
	// NOTE: need check for 2 Kokkos flags?
	// Kokkos doesn't yet support half from full?

	if (irq->ghost != jrq->ghost) continue;		if (irq->ghost != jrq->ghost) continue;
	if (irq->size != jrq->size) continue;		if (irq->size != jrq->size) continue;