From nobody Mon Apr 6 10:42:17 2026 Received: from mail-wr1-f42.google.com (mail-wr1-f42.google.com [209.85.221.42]) (using TLSv1.2 with cipher ECDHE-RSA-AES128-GCM-SHA256 (128/128 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id BAEF13F9F4F for ; Thu, 19 Mar 2026 20:25:38 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.221.42 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1773951943; cv=none; b=WEVjDWYpR0ctlqrjT1OFX7vcK8nIGh/EoZ84x5D+8d5PtyLbPjENnKFD/l4RlSHxVGyX5+JbYLK8BlFLegjEbQ+m9nNMSKARCVTr/A4KwUbBVn4iDAx/6o1H3XGPieSDF+oRan+IxdeMArKG8iqY2nB+JsJyeKdO7UozBndXSqY= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1773951943; c=relaxed/simple; bh=JSyEBFM7jldu6kjAQplAK3yqus2O6kfa1DeMHqQHv4U=; h=From:To:Cc:Subject:Date:Message-ID:In-Reply-To:References: MIME-Version; b=cISYViVaivs53iI84zKZFaOVuJp7kM2Pqp34tLxlVzJxcshd2axmfAxHxFBpUMQOBaMwEQa02ZnQ/24AosmNqGJlhZVNdv2zN6LUhij4qXmPRbOCB/phvNuXmqqckGvYDaXPTVrJ7H/KgVpLNXzdvX0+Gvo27CK5ggyLQPgSKNk= ARC-Authentication-Results: i=1; smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=gmail.com; spf=pass smtp.mailfrom=gmail.com; dkim=pass (2048-bit key) header.d=gmail.com header.i=@gmail.com header.b=b6wRcKbE; arc=none smtp.client-ip=209.85.221.42 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=gmail.com Authentication-Results: smtp.subspace.kernel.org; spf=pass smtp.mailfrom=gmail.com Authentication-Results: smtp.subspace.kernel.org; dkim=pass (2048-bit key) header.d=gmail.com header.i=@gmail.com header.b="b6wRcKbE" Received: by mail-wr1-f42.google.com with SMTP id ffacd0b85a97d-439b97a8a8cso1312825f8f.1 for ; Thu, 19 Mar 2026 13:25:38 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1773951937; x=1774556737; darn=vger.kernel.org; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc:subject:date :message-id:reply-to; bh=SXo/15R90aCs7ovxarZsZGx/BAt8UKSFyqYHwOlOjpI=; b=b6wRcKbEut0E8iDYg9vSckFTr04txJEtyAWlo3qOqD7M0ieAnqiIRzJr5an3dbSvaL K5pkcxUd5/Y5NnAoe0ROFC3u3v1FwkuHujShvHTGn2PGPUYpRTMom7BdPc6i8wJezh2G GoL2sxCcS40W5p6hxStIprkkyfOnmayOkS+ZXPLb2e0/H+m+dEBt+pFsBeNkwYxpZi0v cdsxx0/aO+wH9kP85jEq1PGYqN1z4NXFwAflErock1PjY+7pAmnhZWU9deEE9gMuK/TE 1HIIoKi3F8nQtxkMfGWnBQlOXMyUMRy8nAcrODBjn2IyE7lpERZTmJUaeIy+oVmVZiGo lE0A== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20251104; t=1773951937; x=1774556737; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-gg:x-gm-message-state:from :to:cc:subject:date:message-id:reply-to; bh=SXo/15R90aCs7ovxarZsZGx/BAt8UKSFyqYHwOlOjpI=; b=ZuQMmrwvCn2zp3FcEwPWop0YFbXZR7VkkYx4aSQteWkZEYkrQ1YTfLTHB2T1WWT6q9 9fqFMiAvvOTX0ytrPb5S0eGGpMzL27xID77CwVO0vzqvy5KbXt1iPoXX4zohKolFsJwV LWBdtxPmr5ziYU+c1EX0WkEEk+xz59MM5X6fj7YtWMTjarFM3KBQgLxbjYssi//Y2B/b RYJDzk3c0rhK2+Czn5QJZuDY5V4WRq/ff7Iiblw22mRAQNNeQUyfk/YDn10bB4VzdXw9 t8dmlJXauGYEyyptbizdNfaao4T655xju3e2ZGFRLMY9GYb2BWVV3qzessgbtg0x4ATK MsXw== X-Gm-Message-State: AOJu0YykfTPFJoWAxsAV65BaSHzFSWNNfLMZSZVUCjKr4ScTrpAvtzl8 4BuEMUA/9JhoyrdcxSafuH9B6hYQhgcNy1yGXSYCjKQuCbMbJVO+QiiJ X-Gm-Gg: ATEYQzyQ3VEUW7xRDaHqe35X3pnB4jUZzBKS4mRex7OUQEJv9j5llm1q0HcltV9ecxe ZX232X2OvYMwFVtLASDjynaUhcm5jGtJ2gFLjymvmmvSFcMVLJJl7BeIxYoRxQ4Ds5PLZJvKiCR On+SVbLzhcams90J2oZG6/rlEHNZravC7BxxXHD53A+3obL0xTtrlfSMc7m17Zp3pQSv9To1au3 /IbE9A2gWPtVvAZfUD4Slhmd7xFpzKBgaLKzUrf5whv41LT6vFPhknQAqz+XsTTEmfCqLbbopnf 7+w6vZHLDvhG7qs77FY26cs+RkM+4eECfYghOyeLatW3OsT/Dif32AIer1dYq5lEQBSbt6Bh8dM 2Kb0Bx0Det2Lfowodtyc/p2jlWq0m9eVzYwkASVpNup0KxFyjUlWi1rT0HBWJbwWn/eYnzSkKfX 55qZ161T6a/GQEhcqqRbm7g+IP5qD3dVZ5C+ngu1DyHFyaVwXj X-Received: by 2002:a05:6000:24c3:b0:439:b60a:b400 with SMTP id ffacd0b85a97d-43b64263f07mr1140495f8f.31.1773951936809; Thu, 19 Mar 2026 13:25:36 -0700 (PDT) Received: from LQ5W56KC4T ([2001:8a0:672f:7800:e0e1:55cd:f0b:b1e5]) by smtp.gmail.com with ESMTPSA id ffacd0b85a97d-43b644ae16fsm1347544f8f.8.2026.03.19.13.25.35 (version=TLS1_3 cipher=TLS_CHACHA20_POLY1305_SHA256 bits=256/256); Thu, 19 Mar 2026 13:25:36 -0700 (PDT) From: Eric Curtin X-Google-Original-From: Eric Curtin To: linux-hyperv@vger.kernel.org Cc: linux-kernel@vger.kernel.org, iourit@linux.microsoft.com, wei.liu@kernel.org, decui@microsoft.com, haiyangz@microsoft.com Subject: [PATCH 20/55] drivers: hv: dxgkrnl: Query video memory information Date: Thu, 19 Mar 2026 20:24:34 +0000 Message-ID: <20260319202509.63802-21-eric.curtin@docker.com> X-Mailer: git-send-email 2.51.0 In-Reply-To: <20260319202509.63802-1-eric.curtin@docker.com> References: <20260319202509.63802-1-eric.curtin@docker.com> Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 Content-Transfer-Encoding: quoted-printable Content-Type: text/plain; charset="utf-8" From: Iouri Tarassov Implement the ioctl to query video memory information from the host (LX_DXQUERYVIDEOMEMORYINFO). Signed-off-by: Iouri Tarassov [kms: forward port to 6.6 from 6.1. No code changes made.] Signed-off-by: Kelsey Steele --- drivers/hv/dxgkrnl/dxgkrnl.h | 5 +++ drivers/hv/dxgkrnl/dxgvmbus.c | 64 +++++++++++++++++++++++++++++++++++ drivers/hv/dxgkrnl/dxgvmbus.h | 14 ++++++++ drivers/hv/dxgkrnl/ioctl.c | 50 ++++++++++++++++++++++++++- include/uapi/misc/d3dkmthk.h | 13 +++++++ 5 files changed, 145 insertions(+), 1 deletion(-) diff --git a/drivers/hv/dxgkrnl/dxgkrnl.h b/drivers/hv/dxgkrnl/dxgkrnl.h index ced9dd294f5f..b6a7288a4177 100644 --- a/drivers/hv/dxgkrnl/dxgkrnl.h +++ b/drivers/hv/dxgkrnl/dxgkrnl.h @@ -894,6 +894,11 @@ int dxgvmb_send_query_alloc_residency(struct dxgproces= s *process, struct dxgadapter *adapter, struct d3dkmt_queryallocationresidency *args); +int dxgvmb_send_query_vidmem_info(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dkmt_queryvideomemoryinfo *args, + struct d3dkmt_queryvideomemoryinfo + *__user iargs); int dxgvmb_send_get_device_state(struct dxgprocess *process, struct dxgadapter *adapter, struct d3dkmt_getdevicestate *args, diff --git a/drivers/hv/dxgkrnl/dxgvmbus.c b/drivers/hv/dxgkrnl/dxgvmbus.c index 928fad5f133b..48ff49456057 100644 --- a/drivers/hv/dxgkrnl/dxgvmbus.c +++ b/drivers/hv/dxgkrnl/dxgvmbus.c @@ -1925,6 +1925,70 @@ int dxgvmb_send_query_alloc_residency(struct dxgproc= ess *process, return ret; } =20 +int dxgvmb_send_query_vidmem_info(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dkmt_queryvideomemoryinfo *args, + struct d3dkmt_queryvideomemoryinfo *__user + output) +{ + int ret; + struct dxgkvmb_command_queryvideomemoryinfo *command; + struct dxgkvmb_command_queryvideomemoryinfo_return result =3D { }; + struct dxgvmbusmsg msg =3D {.hdr =3D NULL}; + + ret =3D init_message(&msg, adapter, process, sizeof(*command)); + if (ret) + goto cleanup; + command =3D (void *)msg.msg; + command_vgpu_to_host_init2(&command->hdr, + dxgk_vmbcommand_queryvideomemoryinfo, + process->host_handle); + command->adapter =3D args->adapter; + command->memory_segment_group =3D args->memory_segment_group; + command->physical_adapter_index =3D args->physical_adapter_index; + + ret =3D dxgvmb_send_sync_msg(msg.channel, msg.hdr, msg.size, + &result, sizeof(result)); + if (ret < 0) + goto cleanup; + + ret =3D copy_to_user(&output->budget, &result.budget, + sizeof(output->budget)); + if (ret) { + pr_err("%s failed to copy budget", __func__); + ret =3D -EINVAL; + goto cleanup; + } + ret =3D copy_to_user(&output->current_usage, &result.current_usage, + sizeof(output->current_usage)); + if (ret) { + pr_err("%s failed to copy current usage", __func__); + ret =3D -EINVAL; + goto cleanup; + } + ret =3D copy_to_user(&output->current_reservation, + &result.current_reservation, + sizeof(output->current_reservation)); + if (ret) { + pr_err("%s failed to copy reservation", __func__); + ret =3D -EINVAL; + goto cleanup; + } + ret =3D copy_to_user(&output->available_for_reservation, + &result.available_for_reservation, + sizeof(output->available_for_reservation)); + if (ret) { + pr_err("%s failed to copy avail reservation", __func__); + ret =3D -EINVAL; + } + +cleanup: + free_message(&msg, process); + if (ret) + dev_dbg(DXGDEV, "err: %d", ret); + return ret; +} + int dxgvmb_send_get_device_state(struct dxgprocess *process, struct dxgadapter *adapter, struct d3dkmt_getdevicestate *args, diff --git a/drivers/hv/dxgkrnl/dxgvmbus.h b/drivers/hv/dxgkrnl/dxgvmbus.h index d232eb234e2c..a1549983d50f 100644 --- a/drivers/hv/dxgkrnl/dxgvmbus.h +++ b/drivers/hv/dxgkrnl/dxgvmbus.h @@ -664,6 +664,20 @@ struct dxgkvmb_command_queryallocationresidency_return= { /* d3dkmt_allocationresidencystatus[NumAllocations] */ }; =20 +struct dxgkvmb_command_queryvideomemoryinfo { + struct dxgkvmb_command_vgpu_to_host hdr; + struct d3dkmthandle adapter; + enum d3dkmt_memory_segment_group memory_segment_group; + u32 physical_adapter_index; +}; + +struct dxgkvmb_command_queryvideomemoryinfo_return { + u64 budget; + u64 current_usage; + u64 current_reservation; + u64 available_for_reservation; +}; + struct dxgkvmb_command_getdevicestate { struct dxgkvmb_command_vgpu_to_host hdr; struct d3dkmt_getdevicestate args; diff --git a/drivers/hv/dxgkrnl/ioctl.c b/drivers/hv/dxgkrnl/ioctl.c index 8b7d00e4c881..e692b127e219 100644 --- a/drivers/hv/dxgkrnl/ioctl.c +++ b/drivers/hv/dxgkrnl/ioctl.c @@ -3547,6 +3547,54 @@ dxgkio_flush_heap_transitions(struct dxgprocess *pro= cess, void *__user inargs) return ret; } =20 +static int +dxgkio_query_vidmem_info(struct dxgprocess *process, void *__user inargs) +{ + struct d3dkmt_queryvideomemoryinfo args; + int ret; + struct dxgadapter *adapter =3D NULL; + bool adapter_locked =3D false; + + ret =3D copy_from_user(&args, inargs, sizeof(args)); + if (ret) { + DXG_ERR("failed to copy input args"); + ret =3D -EINVAL; + goto cleanup; + } + + if (args.process !=3D 0) { + DXG_ERR("query vidmem info from another process"); + ret =3D -EINVAL; + goto cleanup; + } + + adapter =3D dxgprocess_adapter_by_handle(process, args.adapter); + if (adapter =3D=3D NULL) { + ret =3D -EINVAL; + goto cleanup; + } + + ret =3D dxgadapter_acquire_lock_shared(adapter); + if (ret < 0) { + adapter =3D NULL; + goto cleanup; + } + adapter_locked =3D true; + + args.adapter =3D adapter->host_handle; + ret =3D dxgvmb_send_query_vidmem_info(process, adapter, &args, inargs); + +cleanup: + + if (adapter_locked) + dxgadapter_release_lock_shared(adapter); + if (adapter) + kref_put(&adapter->adapter_kref, dxgadapter_release); + if (ret < 0) + DXG_ERR("failed: %x", ret); + return ret; +} + static int dxgkio_get_device_state(struct dxgprocess *process, void *__user inargs) { @@ -4287,7 +4335,7 @@ static struct ioctl_desc ioctls[] =3D { /* 0x07 */ {dxgkio_create_paging_queue, LX_DXCREATEPAGINGQUEUE}, /* 0x08 */ {}, /* 0x09 */ {dxgkio_query_adapter_info, LX_DXQUERYADAPTERINFO}, -/* 0x0a */ {}, +/* 0x0a */ {dxgkio_query_vidmem_info, LX_DXQUERYVIDEOMEMORYINFO}, /* 0x0b */ {}, /* 0x0c */ {}, /* 0x0d */ {}, diff --git a/include/uapi/misc/d3dkmthk.h b/include/uapi/misc/d3dkmthk.h index 873feb951129..b7d8b1d91cfc 100644 --- a/include/uapi/misc/d3dkmthk.h +++ b/include/uapi/misc/d3dkmthk.h @@ -897,6 +897,17 @@ enum d3dkmt_memory_segment_group { _D3DKMT_MEMORY_SEGMENT_GROUP_NON_LOCAL =3D 1 }; =20 +struct d3dkmt_queryvideomemoryinfo { + __u64 process; + struct d3dkmthandle adapter; + enum d3dkmt_memory_segment_group memory_segment_group; + __u64 budget; + __u64 current_usage; + __u64 current_reservation; + __u64 available_for_reservation; + __u32 physical_adapter_index; +}; + struct d3dkmt_adaptertype { union { struct { @@ -1204,6 +1215,8 @@ struct d3dkmt_shareobjectwithhost { _IOWR(0x47, 0x07, struct d3dkmt_createpagingqueue) #define LX_DXQUERYADAPTERINFO \ _IOWR(0x47, 0x09, struct d3dkmt_queryadapterinfo) +#define LX_DXQUERYVIDEOMEMORYINFO \ + _IOWR(0x47, 0x0a, struct d3dkmt_queryvideomemoryinfo) #define LX_DXGETDEVICESTATE \ _IOWR(0x47, 0x0e, struct d3dkmt_getdevicestate) #define LX_DXSUBMITCOMMAND \