Skip to content

Commit 8fdd193

Browse files
improve stub for zetKernelGetProfileInfo with ULT
Signed-off-by: davidoli <[email protected]>
1 parent c0268ab commit 8fdd193

File tree

4 files changed

+24
-2
lines changed

4 files changed

+24
-2
lines changed

level_zero/api/tools/zet_module.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
/*
2-
* Copyright (C) 2020 Intel Corporation
2+
* Copyright (C) 2020-2021 Intel Corporation
33
*
44
* SPDX-License-Identifier: MIT
55
*
@@ -21,5 +21,5 @@ ZE_APIEXPORT ze_result_t ZE_APICALL
2121
zetKernelGetProfileInfo(
2222
zet_kernel_handle_t hKernel,
2323
zet_profile_properties_t *pProfileProperties) {
24-
return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
24+
return L0::Kernel::fromHandle(hKernel)->getProfileInfo(pProfileProperties);
2525
}

level_zero/core/source/kernel/kernel.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
#include "shared/source/unified_memory/unified_memory.h"
1414

1515
#include <level_zero/ze_api.h>
16+
#include <level_zero/zet_api.h>
1617

1718
#include <memory>
1819
#include <vector>
@@ -118,6 +119,8 @@ struct Kernel : _ze_kernel_handle_t, virtual NEO::DispatchKernelEncoderI {
118119
virtual ze_result_t suggestMaxCooperativeGroupCount(uint32_t *totalGroupCount) = 0;
119120
virtual ze_result_t setCacheConfig(ze_cache_config_flags_t flags) = 0;
120121

122+
virtual ze_result_t getProfileInfo(zet_profile_properties_t *pProfileProperties) = 0;
123+
121124
virtual const KernelImmutableData *getImmutableData() const = 0;
122125
virtual std::unique_ptr<Kernel> clone() const = 0;
123126

level_zero/core/source/kernel/kernel_imp.h

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,12 @@ struct KernelImp : Kernel {
124124

125125
ze_result_t setCacheConfig(ze_cache_config_flags_t flags) override;
126126

127+
ze_result_t getProfileInfo(zet_profile_properties_t *pProfileProperties) override {
128+
pProfileProperties->flags = 0;
129+
pProfileProperties->numTokens = 0;
130+
return ZE_RESULT_SUCCESS;
131+
}
132+
127133
bool hasIndirectAccess() {
128134
return kernelHasIndirectAccess;
129135
}

level_zero/core/test/unit_tests/sources/kernel/test_kernel.cpp

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -705,6 +705,19 @@ HWTEST_F(KernelPropertiesTests, givenValidKernelThenPropertiesAreRetrieved) {
705705
sizeof(kernelProperties.uuid.mid)));
706706
}
707707

708+
HWTEST_F(KernelPropertiesTests, givenValidKernelThenProfilePropertiesAreRetrieved) {
709+
zet_profile_properties_t kernelProfileProperties = {};
710+
711+
kernelProfileProperties.flags = std::numeric_limits<uint32_t>::max();
712+
kernelProfileProperties.numTokens = std::numeric_limits<uint32_t>::max();
713+
714+
ze_result_t res = kernel->getProfileInfo(&kernelProfileProperties);
715+
EXPECT_EQ(ZE_RESULT_SUCCESS, res);
716+
717+
EXPECT_EQ(0U, kernelProfileProperties.flags);
718+
EXPECT_EQ(0U, kernelProfileProperties.numTokens);
719+
}
720+
708721
HWTEST_F(KernelPropertiesTests, whenSettingValidKernelIndirectAccessFlagsThenFlagsAreSetCorrectly) {
709722
UnifiedMemoryControls unifiedMemoryControls = kernel->getUnifiedMemoryControls();
710723
EXPECT_EQ(false, unifiedMemoryControls.indirectDeviceAllocationsAllowed);

0 commit comments

Comments
 (0)