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.

Reviewers

Commits

Summary

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

Use int8_t instead of bool type.

Looks good.

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

This revision was automatically updated to reflect the committed changes.

Path

Size

libomptarget/

deviceRTLs/

nvptx/

src/

3 lines

5 lines

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

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();
		}