mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-30 09:58:55 +08:00
If a kernel has ray tracing calls, we allocate and initialize per-device RTDispatchGlobals if needed, and hand off pointer to the same into a running kernel via an implicit parameter. Related-To: NEO-5384 Signed-off-by: Jim Snow <jim.m.snow@intel.com>
82 lines
3.5 KiB
C++
82 lines
3.5 KiB
C++
/*
|
|
* Copyright (C) 2021 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/device/device.h"
|
|
#include "shared/test/common/fixtures/device_fixture.h"
|
|
#include "shared/test/common/helpers/default_hw_info.h"
|
|
#include "shared/test/common/helpers/variable_backup.h"
|
|
#include "shared/test/common/mocks/mock_device.h"
|
|
#include "shared/test/common/mocks/ult_device_factory.h"
|
|
|
|
#include "test.h"
|
|
|
|
using namespace NEO;
|
|
|
|
TEST(DeviceBlitterTest, whenBlitterOperationsSupportIsDisabledThenNoInternalCopyEngineIsReturned) {
|
|
VariableBackup<HardwareInfo> backupHwInfo(defaultHwInfo.get());
|
|
defaultHwInfo->capabilityTable.blitterOperationsSupported = false;
|
|
|
|
UltDeviceFactory factory{1, 0};
|
|
EXPECT_EQ(nullptr, factory.rootDevices[0]->getInternalCopyEngine());
|
|
}
|
|
|
|
TEST(DeviceBlitterTest, givenBlitterOperationsDisabledWhenCreatingBlitterEngineThenAbort) {
|
|
VariableBackup<HardwareInfo> backupHwInfo(defaultHwInfo.get());
|
|
defaultHwInfo->capabilityTable.blitterOperationsSupported = false;
|
|
|
|
UltDeviceFactory factory{1, 0};
|
|
EXPECT_THROW(factory.rootDevices[0]->createEngine(0, {aub_stream::EngineType::ENGINE_BCS, EngineUsage::Regular}), std::runtime_error);
|
|
EXPECT_THROW(factory.rootDevices[0]->createEngine(0, {aub_stream::EngineType::ENGINE_BCS, EngineUsage::Cooperative}), std::runtime_error);
|
|
EXPECT_THROW(factory.rootDevices[0]->createEngine(0, {aub_stream::EngineType::ENGINE_BCS, EngineUsage::Internal}), std::runtime_error);
|
|
EXPECT_THROW(factory.rootDevices[0]->createEngine(0, {aub_stream::EngineType::ENGINE_BCS, EngineUsage::LowPriority}), std::runtime_error);
|
|
}
|
|
|
|
using DeviceTest = Test<DeviceFixture>;
|
|
|
|
TEST_F(DeviceTest, whenInitializeRayTracingIsCalledAndRtBackedBufferIsNullptrThenMemoryBackedBufferIsCreated) {
|
|
EXPECT_EQ(nullptr, pDevice->getRTMemoryBackedBuffer());
|
|
EXPECT_EQ(false, pDevice->rayTracingIsInitialized());
|
|
pDevice->initializeRayTracing(0);
|
|
EXPECT_NE(nullptr, pDevice->getRTMemoryBackedBuffer());
|
|
EXPECT_EQ(true, pDevice->rayTracingIsInitialized());
|
|
pDevice->initializeRayTracing(0);
|
|
EXPECT_NE(nullptr, pDevice->getRTMemoryBackedBuffer());
|
|
EXPECT_EQ(true, pDevice->rayTracingIsInitialized());
|
|
}
|
|
|
|
TEST_F(DeviceTest, whenGetRTDispatchGlobalsIsCalledWithUnsupportedBVHLevelsThenNullptrIsReturned) {
|
|
pDevice->initializeRayTracing(5);
|
|
EXPECT_EQ(nullptr, pDevice->getRTDispatchGlobals(100));
|
|
}
|
|
|
|
TEST_F(DeviceTest, whenInitializeRayTracingIsCalledWithMockAllocatorThenRTDispatchGlobalsIsAllocated) {
|
|
pDevice->setRTDispatchGlobalsForceAllocation();
|
|
pDevice->initializeRayTracing(5);
|
|
EXPECT_NE(nullptr, pDevice->getRTDispatchGlobals(3));
|
|
EXPECT_NE(nullptr, pDevice->getRTDispatchGlobals(3));
|
|
EXPECT_NE(nullptr, pDevice->getRTDispatchGlobals(5));
|
|
}
|
|
|
|
TEST_F(DeviceTest, whenInitializeRayTracingIsCalledMultipleTimesWithMockAllocatorThenInitializeRayTracingIsIdempotent) {
|
|
pDevice->setRTDispatchGlobalsForceAllocation();
|
|
pDevice->initializeRayTracing(5);
|
|
EXPECT_NE(nullptr, pDevice->getRTDispatchGlobals(5));
|
|
pDevice->initializeRayTracing(5);
|
|
EXPECT_NE(nullptr, pDevice->getRTDispatchGlobals(5));
|
|
}
|
|
|
|
TEST_F(DeviceTest, whenGetRTDispatchGlobalsIsCalledBeforeInitializationThenNullPtrIsReturned) {
|
|
EXPECT_EQ(nullptr, pDevice->getRTDispatchGlobals(1));
|
|
}
|
|
|
|
TEST_F(DeviceTest, whenGetRTDispatchGlobalsIsCalledWithZeroSizeAndMockAllocatorThenDispatchGlobalsIsReturned) {
|
|
pDevice->setRTDispatchGlobalsForceAllocation();
|
|
EXPECT_EQ(nullptr, pDevice->getRTDispatchGlobals(0));
|
|
pDevice->initializeRayTracing(5);
|
|
EXPECT_NE(nullptr, pDevice->getRTDispatchGlobals(0));
|
|
}
|