99 lines
3.6 KiB
C++
99 lines
3.6 KiB
C++
/*
|
|
* Copyright (C) 2019-2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/compiler_interface/compiler_interface.h"
|
|
#include "shared/source/compiler_interface/compiler_interface.inl"
|
|
#include "shared/test/common/mocks/mock_cif.h"
|
|
#include "shared/test/common/mocks/mock_compilers.h"
|
|
|
|
#include "opencl/test/unit_test/mocks/mock_cl_device.h"
|
|
#include "opencl/test/unit_test/mocks/mock_program.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include <memory>
|
|
|
|
using namespace NEO;
|
|
|
|
struct UpdateSpecConstantsTest : public ::testing::Test {
|
|
void SetUp() override {
|
|
mockProgram.reset(new MockProgram(toClDeviceVector(device)));
|
|
|
|
mockProgram->specConstantsIds.reset(new MockCIFBuffer());
|
|
mockProgram->specConstantsSizes.reset(new MockCIFBuffer());
|
|
|
|
mockProgram->specConstantsIds->PushBackRawCopy(id1);
|
|
mockProgram->specConstantsIds->PushBackRawCopy(id2);
|
|
mockProgram->specConstantsIds->PushBackRawCopy(id3);
|
|
|
|
uint32_t size1 = sizeof(char), size2 = sizeof(uint16_t), size3 = sizeof(int);
|
|
mockProgram->specConstantsSizes->PushBackRawCopy(size1);
|
|
mockProgram->specConstantsSizes->PushBackRawCopy(size2);
|
|
mockProgram->specConstantsSizes->PushBackRawCopy(size3);
|
|
|
|
mockProgram->specConstantsValues.insert({id1, static_cast<uint64_t>(val1)});
|
|
mockProgram->specConstantsValues.insert({id2, static_cast<uint64_t>(val2)});
|
|
mockProgram->specConstantsValues.insert({id3, static_cast<uint64_t>(val3)});
|
|
|
|
values = &mockProgram->specConstantsValues;
|
|
|
|
EXPECT_EQ(val1, static_cast<char>(values->at(id1)));
|
|
EXPECT_EQ(val2, static_cast<uint16_t>(values->at(id2)));
|
|
EXPECT_EQ(val3, static_cast<int>(values->at(id3)));
|
|
}
|
|
MockClDevice device{new MockDevice()};
|
|
std::unique_ptr<MockProgram> mockProgram;
|
|
|
|
uint32_t id1 = 1u;
|
|
uint32_t id2 = 2u;
|
|
uint32_t id3 = 3u;
|
|
char val1 = 5;
|
|
uint16_t val2 = 50;
|
|
int val3 = 500;
|
|
specConstValuesMap *values;
|
|
};
|
|
|
|
TEST_F(UpdateSpecConstantsTest, givenNewSpecConstValueWhenUpdateSpecializationConstantThenProperValueIsCopiedAndUpdated) {
|
|
int newSpecConstVal3 = 5000;
|
|
|
|
auto ret = mockProgram->updateSpecializationConstant(3, sizeof(int), &newSpecConstVal3);
|
|
|
|
EXPECT_EQ(CL_SUCCESS, ret);
|
|
EXPECT_EQ(val1, static_cast<char>(values->at(id1)));
|
|
EXPECT_EQ(val2, static_cast<uint16_t>(values->at(id2)));
|
|
EXPECT_EQ(newSpecConstVal3, static_cast<int>(values->at(id3)));
|
|
|
|
newSpecConstVal3 = 50000;
|
|
EXPECT_NE(newSpecConstVal3, static_cast<int>(values->at(id3)));
|
|
|
|
ret = mockProgram->updateSpecializationConstant(3, sizeof(int), &newSpecConstVal3);
|
|
EXPECT_EQ(CL_SUCCESS, ret);
|
|
EXPECT_EQ(newSpecConstVal3, static_cast<int>(values->at(id3)));
|
|
}
|
|
|
|
TEST_F(UpdateSpecConstantsTest, givenNewSpecConstValueWithUnproperSizeWhenUpdateSpecializationConstantThenErrorIsReturned) {
|
|
int newSpecConstVal3 = 5000;
|
|
|
|
auto ret = mockProgram->updateSpecializationConstant(3, 10 * sizeof(int), &newSpecConstVal3);
|
|
|
|
EXPECT_EQ(CL_INVALID_VALUE, ret);
|
|
EXPECT_EQ(val1, static_cast<char>(values->at(id1)));
|
|
EXPECT_EQ(val2, static_cast<uint16_t>(values->at(id2)));
|
|
EXPECT_EQ(val3, static_cast<int>(values->at(id3)));
|
|
}
|
|
|
|
TEST_F(UpdateSpecConstantsTest, givenNewSpecConstValueWithUnproperIdAndSizeWhenUpdateSpecializationConstantThenErrorIsReturned) {
|
|
int newSpecConstVal3 = 5000;
|
|
|
|
auto ret = mockProgram->updateSpecializationConstant(4, sizeof(int), &newSpecConstVal3);
|
|
|
|
EXPECT_EQ(CL_INVALID_SPEC_ID, ret);
|
|
EXPECT_EQ(val1, static_cast<char>(values->at(id1)));
|
|
EXPECT_EQ(val2, static_cast<uint16_t>(values->at(id2)));
|
|
EXPECT_EQ(val3, static_cast<int>(values->at(id3)));
|
|
}
|