This is an archive of the discontinued LLVM Phabricator instance.

	Show First 20 Lines • Show All 2,617 Lines • ▼ Show 20 Lines

	void __tgt_rtl_data_alloc(int DeviceId, int64_t Size, void , int32_t Kind) {			void __tgt_rtl_data_alloc(int DeviceId, int64_t Size, void , int32_t Kind) {
	void *Ptr = NULL;			void *Ptr = NULL;
	assert(DeviceId < DeviceInfo().NumberOfDevices && "Device ID too large");			assert(DeviceId < DeviceInfo().NumberOfDevices && "Device ID too large");

	hsa_amd_memory_pool_t MemoryPool;			hsa_amd_memory_pool_t MemoryPool;
	switch (Kind) {			switch (Kind) {
	case TARGET_ALLOC_DEFAULT:			case TARGET_ALLOC_DEFAULT:
				case TARGET_ALLOC_DEVICE:
	// GPU memory			// GPU memory
	MemoryPool = DeviceInfo().getDeviceMemoryPool(DeviceId);			MemoryPool = DeviceInfo().getDeviceMemoryPool(DeviceId);
	break;			break;
	case TARGET_ALLOC_HOST:			case TARGET_ALLOC_HOST:
	// non-migratable memory accessible by host and device(s)			// non-migratable memory accessible by host and device(s)
	MemoryPool = DeviceInfo().getHostMemoryPool();			MemoryPool = DeviceInfo().getHostMemoryPool();
	break;			break;
	default:			default:
	▲ Show 20 Lines • Show All 140 Lines • Show Last 20 Lines

openmp/libomptarget/test/api/omp_device_memory.c

	// RUN: %libomptarget-compile-run-and-check-nvptx64-nvidia-cuda			// RUN: %libomptarget-compile-run-and-check-generic
	// REQUIRES: nvptx64-nvidia-cuda

	#include <omp.h>			#include <omp.h>
	#include <stdio.h>			#include <stdio.h>

	int main() {			int main() {
	const int N = 64;			const int N = 64;

	int *device_ptr =			int *device_ptr =
	Show All 18 Lines