mirror of
https://github.com/blender/blender
synced 2026-09-26 16:15:47 +03:00
7026d9ac43
* Add HIP-RT API functions and library loading * Add more HIP API types and functions * Find HIP linker executable in CMake module * New CMake module to find HIP-RT SDK Co-authored-by: Brecht Van Lommel <brecht@blender.org> Ref #105538
97 lines
2.6 KiB
C++
97 lines
2.6 KiB
C++
/*
|
|
* Copyright 2011-2023 Blender Foundation
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License
|
|
*/
|
|
|
|
#include "util.h"
|
|
|
|
#include <hiprtew.h>
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
|
|
static DynamicLibrary hiprt_lib;
|
|
|
|
#define HIPRT_LIBRARY_FIND(name) \
|
|
name = (t##name *)dynamic_library_find(hiprt_lib, #name);
|
|
|
|
/* Function definitions. */
|
|
thiprtCreateContext *hiprtCreateContext;
|
|
thiprtDestroyContext *hiprtDestroyContext;
|
|
thiprtCreateGeometry *hiprtCreateGeometry;
|
|
thiprtDestroyGeometry *hiprtDestroyGeometry;
|
|
thiprtBuildGeometry *hiprtBuildGeometry;
|
|
thiprtGetGeometryBuildTemporaryBufferSize *hiprtGetGeometryBuildTemporaryBufferSize;
|
|
thiprtCreateScene *hiprtCreateScene;
|
|
thiprtDestroyScene *hiprtDestroyScene;
|
|
thiprtBuildScene *hiprtBuildScene;
|
|
thiprtGetSceneBuildTemporaryBufferSize *hiprtGetSceneBuildTemporaryBufferSize;
|
|
thiprtCreateFuncTable *hiprtCreateFuncTable;
|
|
thiprtSetFuncTable *hiprtSetFuncTable;
|
|
thiprtDestroyFuncTable *hiprtDestroyFuncTable;
|
|
|
|
static void hipewHipRtExit(void)
|
|
{
|
|
if (hiprt_lib != NULL) {
|
|
/* Ignore errors. */
|
|
dynamic_library_close(hiprt_lib);
|
|
hiprt_lib = NULL;
|
|
}
|
|
}
|
|
|
|
bool hiprtewInit()
|
|
{
|
|
static bool result = false;
|
|
static bool initialized = false;
|
|
|
|
if (initialized) {
|
|
return result;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
initialized = true;
|
|
|
|
if (atexit(hipewHipRtExit)) {
|
|
return false;
|
|
}
|
|
|
|
std::string hiprt_ver(HIPRT_VERSION_STR);
|
|
std::string hiprt_path = "hiprt" + hiprt_ver + "64.dll";
|
|
|
|
hiprt_lib = dynamic_library_open(hiprt_path.c_str());
|
|
|
|
if (hiprt_lib == NULL) {
|
|
return false;
|
|
}
|
|
|
|
HIPRT_LIBRARY_FIND(hiprtCreateContext)
|
|
HIPRT_LIBRARY_FIND(hiprtDestroyContext)
|
|
HIPRT_LIBRARY_FIND(hiprtCreateGeometry)
|
|
HIPRT_LIBRARY_FIND(hiprtDestroyGeometry)
|
|
HIPRT_LIBRARY_FIND(hiprtBuildGeometry)
|
|
HIPRT_LIBRARY_FIND(hiprtGetGeometryBuildTemporaryBufferSize)
|
|
HIPRT_LIBRARY_FIND(hiprtCreateScene)
|
|
HIPRT_LIBRARY_FIND(hiprtDestroyScene)
|
|
HIPRT_LIBRARY_FIND(hiprtBuildScene)
|
|
HIPRT_LIBRARY_FIND(hiprtGetSceneBuildTemporaryBufferSize)
|
|
HIPRT_LIBRARY_FIND(hiprtCreateFuncTable)
|
|
HIPRT_LIBRARY_FIND(hiprtSetFuncTable)
|
|
HIPRT_LIBRARY_FIND(hiprtDestroyFuncTable)
|
|
|
|
result = true;
|
|
#endif
|
|
|
|
return result;
|
|
}
|