llvm-project/offload/unittests/OffloadAPI/kernel/olCalculateOptimalOccupancy.cpp
Ross Brunton 2c11a83691
[Offload] Add olCalculateOptimalOccupancy (#142950)
This is equivalent to `cuOccupancyMaxPotentialBlockSize`. It is
currently
only implemented on Cuda; AMDGPU and Host return unsupported.

---------

Co-authored-by: Callum Fare <callum@codeplay.com>
2025-08-19 15:16:47 +01:00

46 lines
1.5 KiB
C++

//===------- Offload API tests - olCalculateOptimalOccupancy --------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#include "../common/Fixtures.hpp"
#include <OffloadAPI.h>
#include <gtest/gtest.h>
using olCalculateOptimalOccupancyTest = OffloadKernelTest;
OFFLOAD_TESTS_INSTANTIATE_DEVICE_FIXTURE(olCalculateOptimalOccupancyTest);
TEST_P(olCalculateOptimalOccupancyTest, Success) {
size_t Size{0};
ASSERT_SUCCESS_OR_UNSUPPORTED(
olCalculateOptimalOccupancy(Device, Kernel, 0, &Size));
ASSERT_GT(Size, 0u);
}
TEST_P(olCalculateOptimalOccupancyTest, SuccessMem) {
size_t Size{0};
ASSERT_SUCCESS_OR_UNSUPPORTED(
olCalculateOptimalOccupancy(Device, Kernel, 1024, &Size));
ASSERT_GT(Size, 0u);
}
TEST_P(olCalculateOptimalOccupancyTest, NullKernel) {
size_t Size;
ASSERT_ERROR(OL_ERRC_INVALID_NULL_HANDLE,
olCalculateOptimalOccupancy(Device, nullptr, 0, &Size));
}
TEST_P(olCalculateOptimalOccupancyTest, NullDevice) {
size_t Size;
ASSERT_ERROR(OL_ERRC_INVALID_NULL_HANDLE,
olCalculateOptimalOccupancy(nullptr, Kernel, 0, &Size));
}
TEST_P(olCalculateOptimalOccupancyTest, NullOutput) {
ASSERT_ERROR(OL_ERRC_INVALID_NULL_POINTER,
olCalculateOptimalOccupancy(Device, Kernel, 0, nullptr));
}