This is an archive of the discontinued LLVM Phabricator instance.

[OpenMP][libomptarget] Add function for checking SPMD mode
ClosedPublic

Authored by gtbercea on May 14 2018, 12:16 PM.

Download Raw Diff

Details

Reviewers

ABataev
grokos
carlo.bertolli
caomhin

Commits

rG787a3500214d: [OpenMP][libomptarget] Add function for checking SPMD mode
rOMP332360: [OpenMP][libomptarget] Add function for checking SPMD mode
rL332360: [OpenMP][libomptarget] Add function for checking SPMD mode

Summary

Add function to the NVPTX libomptarget library that will return true if the current target region is being executed in SPMD mode.

Diff Detail

Repository: rL LLVM

Event Timeline

gtbercea created this revision.May 14 2018, 12:16 PM

Herald added subscribers: openmp-commits, guansong. · View Herald TranscriptMay 14 2018, 12:16 PM

Use int8_t instead of bool type.

Looks good.

This revision is now accepted and ready to land.May 15 2018, 8:12 AM

Closed by commit rL332360: [OpenMP][libomptarget] Add function for checking SPMD mode (authored by gbercea). · Explain WhyMay 15 2018, 8:20 AM

This revision was automatically updated to reflect the committed changes.

Herald added a subscriber: llvm-commits. · View Herald TranscriptMay 15 2018, 8:20 AM

Revision Contents

Path

Size

openmp/

trunk/

libomptarget/

deviceRTLs/

nvptx/

src/

interface.h

3 lines

omptarget-nvptx.cu

5 lines

Diff 146835

openmp/trunk/libomptarget/deviceRTLs/nvptx/src/interface.h

Show First 20 Lines • Show All 510 Lines • ▼ Show 20 Lines	EXTERN void *__kmpc_data_sharing_environment_begin(
int16_t IsOMPRuntimeInitialized);		int16_t IsOMPRuntimeInitialized);
EXTERN void __kmpc_data_sharing_environment_end(		EXTERN void __kmpc_data_sharing_environment_end(
__kmpc_data_sharing_slot SavedSharedSlot, void SavedSharedStack,		__kmpc_data_sharing_slot SavedSharedSlot, void SavedSharedStack,
void *SavedSharedFrame, int32_t SavedActiveThreads, int32_t IsEntryPoint);		void *SavedSharedFrame, int32_t SavedActiveThreads, int32_t IsEntryPoint);

EXTERN void *		EXTERN void *
__kmpc_get_data_sharing_environment_frame(int32_t SourceThreadID,		__kmpc_get_data_sharing_environment_frame(int32_t SourceThreadID,
int16_t IsOMPRuntimeInitialized);		int16_t IsOMPRuntimeInitialized);

		// SPMD execution mode interrogation function.
		EXTERN int8_t __kmpc_is_spmd_exec_mode();
#endif		#endif

openmp/trunk/libomptarget/deviceRTLs/nvptx/src/omptarget-nvptx.cu

Show First 20 Lines • Show All 181 Lines • ▼ Show 20 Lines	EXTERN void __kmpc_spmd_kernel_deinit() {
int threadId = GetThreadIdInBlock();		int threadId = GetThreadIdInBlock();
if (threadId == 0) {		if (threadId == 0) {
// Enqueue omp state object for use by another team.		// Enqueue omp state object for use by another team.
int slot = smid() % MAX_SM;		int slot = smid() % MAX_SM;
omptarget_nvptx_device_State[slot].Enqueue(		omptarget_nvptx_device_State[slot].Enqueue(
omptarget_nvptx_threadPrivateContext);		omptarget_nvptx_threadPrivateContext);
}		}
}		}

		// Return true if the current target region is executed in SPMD mode.
		EXTERN int8_t __kmpc_is_spmd_exec_mode() {
		return isSPMDMode();
		}