2015-04-28 22:46:37 +00:00
|
|
|
//
|
|
|
|
// Copyright 2015 Pixar
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "Apache License")
|
|
|
|
// with the following modification; you may not use this file except in
|
|
|
|
// compliance with the Apache License and the following modification to it:
|
|
|
|
// Section 6. Trademarks. is deleted and replaced with:
|
|
|
|
//
|
|
|
|
// 6. Trademarks. This License does not grant permission to use the trade
|
|
|
|
// names, trademarks, service marks, or product names of the Licensor
|
|
|
|
// and its affiliates, except as required to comply with Section 4(c) of
|
|
|
|
// the License and to reproduce the content of the NOTICE file.
|
|
|
|
//
|
|
|
|
// You may obtain a copy of the Apache License at
|
|
|
|
//
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the Apache License with the above modification is
|
|
|
|
// distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
|
|
// KIND, either express or implied. See the Apache License for the specific
|
|
|
|
// language governing permissions and limitations under the Apache License.
|
|
|
|
//
|
|
|
|
|
|
|
|
#include "cudaDeviceContext.h"
|
|
|
|
|
|
|
|
#if defined(_WIN32)
|
|
|
|
#include <windows.h>
|
|
|
|
#elif defined(__APPLE__)
|
|
|
|
#include <OpenGL/OpenGL.h>
|
|
|
|
#else
|
|
|
|
#include <X11/Xlib.h>
|
|
|
|
#include <GL/glx.h>
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#include <cstdio>
|
|
|
|
#include <cuda.h>
|
|
|
|
#include <cuda_runtime_api.h>
|
|
|
|
#include <cuda_gl_interop.h>
|
|
|
|
|
2015-04-29 18:51:12 +00:00
|
|
|
#if defined(OPENSUBDIV_HAS_DX)
|
|
|
|
#include <cuda_d3d11_interop.h>
|
|
|
|
#endif
|
|
|
|
|
2015-04-28 22:46:37 +00:00
|
|
|
#define message(fmt, ...)
|
|
|
|
//#define message(fmt, ...) fprintf(stderr, fmt, __VA_ARGS__)
|
|
|
|
#define error(fmt, ...) fprintf(stderr, fmt, __VA_ARGS__)
|
|
|
|
|
|
|
|
// -----------------------------------------------------------------------
|
|
|
|
#if CUDA_VERSION < 5000
|
|
|
|
static int _GetCudaDeviceForCurrentGLContext()
|
|
|
|
{
|
|
|
|
#if defined(_WIN32)
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
#elif defined(__APPLE__)
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
#else // X11
|
|
|
|
// If we don't have a current GL context, then choose the device which
|
|
|
|
// matches the current X11 screen number.
|
|
|
|
Display * display = glXGetCurrentDisplay();
|
|
|
|
if (not display) {
|
|
|
|
display = XOpenDisplay(NULL);
|
|
|
|
if (display) {
|
|
|
|
int screen = DefaultScreen(display);
|
|
|
|
XCloseDisplay(display);
|
|
|
|
message("CUDA init using device for default screen: %d\n", screen);
|
|
|
|
return screen;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
// We can't use the new interop API, so use the device
|
|
|
|
// corresponding to the screen number of the current GL context.
|
|
|
|
int screen = DefaultScreen(display);
|
|
|
|
message("CUDA init using device for screen: %d\n", screen);
|
|
|
|
return screen;
|
|
|
|
#endif // X11
|
|
|
|
}
|
|
|
|
|
|
|
|
#else // CUDA_VERSION >= 50000 -----------------------------------------
|
|
|
|
static int _GetCudaDeviceForCurrentGLContext()
|
|
|
|
{
|
|
|
|
// Find and use the CUDA device for the current GL context
|
|
|
|
unsigned int interopDeviceCount = 0;
|
|
|
|
int interopDevices[1];
|
|
|
|
cudaError_t status = cudaGLGetDevices(&interopDeviceCount, interopDevices,
|
|
|
|
1, cudaGLDeviceListCurrentFrame);
|
|
|
|
if (status == cudaErrorNoDevice or interopDeviceCount != 1) {
|
|
|
|
message("CUDA no interop devices found.\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
int device = interopDevices[0];
|
|
|
|
|
|
|
|
#if defined(_WIN32)
|
|
|
|
return device;
|
|
|
|
|
|
|
|
#elif defined(__APPLE__)
|
|
|
|
return device;
|
|
|
|
|
|
|
|
#else // X11
|
|
|
|
Display * display = glXGetCurrentDisplay();
|
|
|
|
int screen = DefaultScreen(display);
|
|
|
|
if (device != screen) {
|
|
|
|
error("The CUDA interop device (%d) does not match "
|
|
|
|
"the screen used by the current GL context (%d), "
|
|
|
|
"which may cause slow performance on systems "
|
|
|
|
"with multiple GPU devices.", device, screen);
|
|
|
|
}
|
|
|
|
message("CUDA init using device for current GL context: %d\n", device);
|
|
|
|
return device;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
#endif // CUDA_VERSION -----------------------------------------------
|
|
|
|
|
|
|
|
CudaDeviceContext::CudaDeviceContext() :
|
|
|
|
_initialized(false) {
|
|
|
|
}
|
|
|
|
|
|
|
|
CudaDeviceContext::~CudaDeviceContext() {
|
|
|
|
cudaDeviceReset();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
CudaDeviceContext::Initialize() {
|
|
|
|
|
|
|
|
// see if any cuda device is available.
|
|
|
|
int deviceCount = 0;
|
|
|
|
cudaGetDeviceCount(&deviceCount);
|
|
|
|
message("CUDA device count: %d\n", deviceCount);
|
|
|
|
if (deviceCount <= 0) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
cudaGLSetGLDevice(_GetCudaDeviceForCurrentGLContext());
|
|
|
|
_initialized = true;
|
|
|
|
return true;
|
|
|
|
}
|
2015-04-29 18:51:12 +00:00
|
|
|
|
|
|
|
bool
|
|
|
|
CudaDeviceContext::Initialize(ID3D11Device *device) {
|
|
|
|
|
|
|
|
#if defined(OPENSUBDIV_HAS_DX)
|
|
|
|
cudaD3D11SetDirect3DDevice(device);
|
|
|
|
return true;
|
|
|
|
#else
|
|
|
|
(void)device; // unused
|
|
|
|
return false;
|
|
|
|
#endif
|
|
|
|
}
|