From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from mgamail.intel.com (mgamail.intel.com [192.55.52.120]) by gabe.freedesktop.org (Postfix) with ESMTPS id 2FE1C10E1DF for ; Mon, 7 Aug 2023 09:24:57 +0000 (UTC) From: Lukasz Laguna To: igt-dev@lists.freedesktop.org Date: Mon, 7 Aug 2023 11:24:46 +0200 Message-Id: <20230807092446.7805-1-lukasz.laguna@intel.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit Subject: [igt-dev] [PATCH i-g-t] lib/xe_query: add helper getting available VRAM List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Cc: mauro.chehab@intel.com Errors-To: igt-dev-bounces@lists.freedesktop.org Sender: "igt-dev" List-ID: Helper allows to read available VRAM of a specified device and GT. v2: - Query for current memory usage instead of using cached one (Zbigniew) v3: - Reacquire mem_region pointer (Zbigniew) v4: - Update value of used memory only in xe_dev Signed-off-by: Lukasz Laguna --- lib/xe/xe_query.c | 35 +++++++++++++++++++++++++++++++++++ lib/xe/xe_query.h | 1 + 2 files changed, 36 insertions(+) diff --git a/lib/xe/xe_query.c b/lib/xe/xe_query.c index 8963c7b06..c356abe1e 100644 --- a/lib/xe/xe_query.c +++ b/lib/xe/xe_query.c @@ -630,6 +630,41 @@ uint64_t xe_visible_vram_size(int fd, int gt) return visible_size; } +/** + * xe_vram_available: + * @fd: xe device fd + * @gt: gt + * + * Returns available vram of xe device @fd and @gt. + */ +uint64_t xe_vram_available(int fd, int gt) +{ + struct xe_device *xe_dev; + int region_idx; + struct drm_xe_query_mem_region *mem_region; + struct drm_xe_query_mem_usage *mem_usage; + + xe_dev = find_in_cache(fd); + igt_assert(xe_dev); + + region_idx = ffs(native_region_for_gt(xe_dev->gts, gt)) - 1; + mem_region = &xe_dev->mem_usage->regions[region_idx]; + + if (XE_IS_CLASS_VRAM(mem_region)) { + uint64_t available_vram; + + mem_usage = xe_query_mem_usage_new(fd); + pthread_mutex_lock(&cache.cache_mutex); + mem_region->used = mem_usage->regions[region_idx].used; + available_vram = mem_region->total_size - mem_region->used; + pthread_mutex_unlock(&cache.cache_mutex); + free(mem_usage); + + return available_vram; + } + + return 0; +} /** * xe_get_default_alignment: diff --git a/lib/xe/xe_query.h b/lib/xe/xe_query.h index 1b74c58ab..20dbfa12c 100644 --- a/lib/xe/xe_query.h +++ b/lib/xe/xe_query.h @@ -97,6 +97,7 @@ unsigned int xe_number_hw_engines(int fd); bool xe_has_vram(int fd); uint64_t xe_vram_size(int fd, int gt); uint64_t xe_visible_vram_size(int fd, int gt); +uint64_t xe_vram_available(int fd, int gt); uint32_t xe_get_default_alignment(int fd); uint32_t xe_va_bits(int fd); uint16_t xe_dev_id(int fd); -- 2.40.0