mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2026-10-11 23:38:25 +08:00
Add and update samples with CUDA 10.1 support
This commit is contained in:
Mahesh Doijade
@@ -282,6 +282,42 @@ static const char *_cudaGetErrorEnum(curandStatus_t error) {
|
||||
}
|
||||
#endif
|
||||
|
||||
#ifdef NVJPEGAPI
|
||||
// nvJPEG API errors
|
||||
static const char *_cudaGetErrorEnum(nvjpegStatus_t error) {
|
||||
switch (error) {
|
||||
case NVJPEG_STATUS_SUCCESS:
|
||||
return "NVJPEG_STATUS_SUCCESS";
|
||||
|
||||
case NVJPEG_STATUS_NOT_INITIALIZED:
|
||||
return "NVJPEG_STATUS_NOT_INITIALIZED";
|
||||
|
||||
case NVJPEG_STATUS_INVALID_PARAMETER:
|
||||
return "NVJPEG_STATUS_INVALID_PARAMETER";
|
||||
|
||||
case NVJPEG_STATUS_BAD_JPEG:
|
||||
return "NVJPEG_STATUS_BAD_JPEG";
|
||||
|
||||
case NVJPEG_STATUS_JPEG_NOT_SUPPORTED:
|
||||
return "NVJPEG_STATUS_JPEG_NOT_SUPPORTED";
|
||||
|
||||
case NVJPEG_STATUS_ALLOCATOR_FAILURE:
|
||||
return "NVJPEG_STATUS_ALLOCATOR_FAILURE";
|
||||
|
||||
case NVJPEG_STATUS_EXECUTION_FAILED:
|
||||
return "NVJPEG_STATUS_EXECUTION_FAILED";
|
||||
|
||||
case NVJPEG_STATUS_ARCH_MISMATCH:
|
||||
return "NVJPEG_STATUS_ARCH_MISMATCH";
|
||||
|
||||
case NVJPEG_STATUS_INTERNAL_ERROR:
|
||||
return "NVJPEG_STATUS_INTERNAL_ERROR";
|
||||
}
|
||||
|
||||
return "<unknown>";
|
||||
}
|
||||
#endif
|
||||
|
||||
#ifdef NV_NPPIDEFS_H
|
||||
// NPP API errors
|
||||
static const char *_cudaGetErrorEnum(NppStatus error) {
|
||||
|
||||
178
Common/helper_multiprocess.cpp
Normal file
178
Common/helper_multiprocess.cpp
Normal file
@@ -0,0 +1,178 @@
|
||||
/* Copyright (c) 2018, NVIDIA CORPORATION. All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
* * Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* * Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in the
|
||||
* documentation and/or other materials provided with the distribution.
|
||||
* * Neither the name of NVIDIA CORPORATION nor the names of its
|
||||
* contributors may be used to endorse or promote products derived
|
||||
* from this software without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
|
||||
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
|
||||
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
|
||||
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include "helper_multiprocess.h"
|
||||
#include <cstdlib>
|
||||
#include <string>
|
||||
|
||||
int sharedMemoryCreate(const char *name, size_t sz, sharedMemoryInfo *info)
|
||||
{
|
||||
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
|
||||
info->size = sz;
|
||||
info->shmHandle = CreateFileMapping(INVALID_HANDLE_VALUE,
|
||||
NULL,
|
||||
PAGE_READWRITE,
|
||||
0,
|
||||
(DWORD)sz,
|
||||
name);
|
||||
if (info->shmHandle == 0) {
|
||||
return GetLastError();
|
||||
}
|
||||
|
||||
info->addr = MapViewOfFile(info->shmHandle, FILE_MAP_ALL_ACCESS, 0, 0, sz);
|
||||
if (info->addr == NULL) {
|
||||
return GetLastError();
|
||||
}
|
||||
|
||||
return 0;
|
||||
#else
|
||||
int status = 0;
|
||||
|
||||
info->size = sz;
|
||||
|
||||
info->shmFd = shm_open(name, O_RDWR | O_CREAT, 0777);
|
||||
if (info->shmFd < 0) {
|
||||
return errno;
|
||||
}
|
||||
|
||||
status = ftruncate(info->shmFd, sz);
|
||||
if (status != 0) {
|
||||
return status;
|
||||
}
|
||||
|
||||
info->addr = mmap(0, sz, PROT_READ | PROT_WRITE, MAP_SHARED, info->shmFd, 0);
|
||||
if (info->addr == NULL) {
|
||||
return errno;
|
||||
}
|
||||
|
||||
return 0;
|
||||
#endif
|
||||
}
|
||||
|
||||
int sharedMemoryOpen(const char *name, size_t sz, sharedMemoryInfo *info)
|
||||
{
|
||||
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
|
||||
info->size = sz;
|
||||
|
||||
info->shmHandle = OpenFileMapping(FILE_MAP_ALL_ACCESS, FALSE, name);
|
||||
if (info->shmHandle == 0) {
|
||||
return GetLastError();
|
||||
}
|
||||
|
||||
info->addr = MapViewOfFile(info->shmHandle, FILE_MAP_ALL_ACCESS, 0, 0, sz);
|
||||
if (info->addr == NULL) {
|
||||
return GetLastError();
|
||||
}
|
||||
|
||||
return 0;
|
||||
#else
|
||||
info->size = sz;
|
||||
|
||||
info->shmFd = shm_open(name, O_RDWR, 0777);
|
||||
if (info->shmFd < 0) {
|
||||
return errno;
|
||||
}
|
||||
|
||||
info->addr = mmap(0, sz, PROT_READ | PROT_WRITE, MAP_SHARED, info->shmFd, 0);
|
||||
if (info->addr == NULL) {
|
||||
return errno;
|
||||
}
|
||||
|
||||
return 0;
|
||||
#endif
|
||||
}
|
||||
|
||||
void sharedMemoryClose(sharedMemoryInfo *info)
|
||||
{
|
||||
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
|
||||
if (info->addr) {
|
||||
UnmapViewOfFile(info->addr);
|
||||
}
|
||||
if (info->shmHandle) {
|
||||
CloseHandle(info->shmHandle);
|
||||
}
|
||||
#else
|
||||
if (info->addr) {
|
||||
munmap(info->addr, info->size);
|
||||
}
|
||||
if (info->shmFd) {
|
||||
close(info->shmFd);
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
int spawnProcess(Process *process, const char *app, char * const *args)
|
||||
{
|
||||
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
|
||||
STARTUPINFO si = {0};
|
||||
BOOL status;
|
||||
size_t arglen = 0;
|
||||
size_t argIdx = 0;
|
||||
std::string arg_string;
|
||||
memset(process, 0, sizeof(*process));
|
||||
|
||||
while (*args) {
|
||||
arg_string.append(*args).append(1, ' ');
|
||||
args++;
|
||||
}
|
||||
|
||||
status = CreateProcess(app, LPSTR(arg_string.c_str()), NULL, NULL, FALSE, 0, NULL, NULL, &si, process);
|
||||
|
||||
return status ? 0 : GetLastError();
|
||||
#else
|
||||
*process = fork();
|
||||
if (*process == 0) {
|
||||
if (0 > execvp(app, args)) {
|
||||
return errno;
|
||||
}
|
||||
}
|
||||
else if (*process < 0) {
|
||||
return errno;
|
||||
}
|
||||
return 0;
|
||||
#endif
|
||||
}
|
||||
|
||||
int waitProcess(Process *process)
|
||||
{
|
||||
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
|
||||
DWORD exitCode;
|
||||
WaitForSingleObject(process->hProcess, INFINITE);
|
||||
GetExitCodeProcess(process->hProcess, &exitCode);
|
||||
CloseHandle(process->hProcess);
|
||||
CloseHandle(process->hThread);
|
||||
return (int)exitCode;
|
||||
#else
|
||||
int status = 0;
|
||||
do {
|
||||
if (0 > waitpid(*process, &status, 0)) {
|
||||
return errno;
|
||||
}
|
||||
} while (!WIFEXITED(status));
|
||||
return WEXITSTATUS(status);
|
||||
#endif
|
||||
}
|
||||
71
Common/helper_multiprocess.h
Normal file
71
Common/helper_multiprocess.h
Normal file
@@ -0,0 +1,71 @@
|
||||
/* Copyright (c) 2018, NVIDIA CORPORATION. All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
* * Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* * Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in the
|
||||
* documentation and/or other materials provided with the distribution.
|
||||
* * Neither the name of NVIDIA CORPORATION nor the names of its
|
||||
* contributors may be used to endorse or promote products derived
|
||||
* from this software without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
|
||||
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
|
||||
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
|
||||
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef HELPER_MULTIPROCESS_H
|
||||
#define HELPER_MULTIPROCESS_H
|
||||
|
||||
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
|
||||
#ifndef WIN32_LEAN_AND_MEAN
|
||||
#define WIN32_LEAN_AND_MEAN
|
||||
#endif
|
||||
#include <windows.h>
|
||||
#else
|
||||
#include <fcntl.h>
|
||||
#include <sys/mman.h>
|
||||
#include <unistd.h>
|
||||
#include <errno.h>
|
||||
#include <sys/wait.h>
|
||||
#endif
|
||||
|
||||
typedef struct sharedMemoryInfo_st {
|
||||
void *addr;
|
||||
size_t size;
|
||||
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
|
||||
HANDLE shmHandle;
|
||||
#else
|
||||
int shmFd;
|
||||
#endif
|
||||
} sharedMemoryInfo;
|
||||
|
||||
int sharedMemoryCreate(const char *name, size_t sz, sharedMemoryInfo *info);
|
||||
|
||||
int sharedMemoryOpen(const char *name, size_t sz, sharedMemoryInfo *info);
|
||||
|
||||
void sharedMemoryClose(sharedMemoryInfo *info);
|
||||
|
||||
|
||||
#if defined(WIN32) || defined(_WIN32) || defined(WIN64) || defined(_WIN64)
|
||||
typedef PROCESS_INFORMATION Process;
|
||||
#else
|
||||
typedef pid_t Process;
|
||||
#endif
|
||||
|
||||
int spawnProcess(Process *process, const char *app, char * const *args);
|
||||
|
||||
int waitProcess(Process *process);
|
||||
|
||||
#endif // HELPER_MULTIPROCESS_H
|
||||
Reference in New Issue
Block a user