Files
Joseph Huber d62cd1b89d [Offload] Add argument to 'olInit' for global configuration options (#181872)
Summary:
This PR adds a pointer argument to the initialization routine to be used
for global options. Right now this is used to allow the user to
constrain which backends they wish to use.

If a null argument is passed, the same behavior as before is observed.
This is epxected to be extensible by forcing the user to encode the size
of the struct. So, old executables will encode which fields they have
access to.

We use a macro helper to get this struct rather than a runtime call so
that the current state of the size is baked into the executable rather
than something looked up by the runtime. Otherwise it would just return
the size that the (potentially newer) runtime would see
2026-02-17 14:04:00 -06:00

50 lines
1.4 KiB
C++

//===------- Offload API tests - olInit -----------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
// NOTE: For this test suite, the implicit olInit/olShutDown doesn't happen, so
// tests have to do it themselves
#include "../common/Fixtures.hpp"
#include <OffloadAPI.h>
#include <gtest/gtest.h>
struct olInitTest : ::testing::Test {};
TEST_F(olInitTest, Success) {
ASSERT_SUCCESS(olInit(nullptr));
ASSERT_SUCCESS(olShutDown());
}
TEST_F(olInitTest, Uninitialized) {
ASSERT_ERROR(OL_ERRC_UNINITIALIZED,
olIterateDevices(
[](ol_device_handle_t, void *) { return false; }, nullptr));
}
TEST_F(olInitTest, RepeatedInit) {
for (size_t I = 0; I < 10; I++) {
ASSERT_SUCCESS(olInit(nullptr));
ASSERT_SUCCESS(olShutDown());
}
}
TEST_F(olInitTest, WithInitArgs) {
ol_init_args_t Args = OL_INIT_ARGS_INIT;
ol_platform_backend_t Backends[] = {OL_PLATFORM_BACKEND_HOST};
Args.NumPlatforms = 1;
Args.Platforms = Backends;
ASSERT_SUCCESS(olInit(&Args));
ASSERT_SUCCESS(olShutDown());
}
TEST_F(olInitTest, InvalidSize) {
ol_init_args_t Args = OL_INIT_ARGS_INIT;
Args.Size = 0;
ASSERT_ERROR(OL_ERRC_INVALID_SIZE, olInit(&Args));
}