From nobody Mon Apr 6 10:41:59 2026 Received: from mail-wr1-f47.google.com (mail-wr1-f47.google.com [209.85.221.47]) (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 AF4503F9F58 for ; Thu, 19 Mar 2026 20:25:36 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.221.47 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1773951942; cv=none; b=En+4NU+2f8amJ32HfotHZ/7KAMmb/Mf9pQw2cDUkUhASMpWVotgenomw0rYNY0cFR516n1/5lmzrQJmO5UOfel1laypLHHuoE75Op6riB97bQDmU2+rZoVYb9ZBnhVFA5uk3OhL/V/dQUR4b2FjlpImExJTCiDTxY9aIqqBCoYo= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1773951942; c=relaxed/simple; bh=3j1wDPqQSZdr8k5MF2F6ulqP/7AQz9mW4yJoIeHwvak=; h=From:To:Cc:Subject:Date:Message-ID:In-Reply-To:References: MIME-Version; b=c55UGOwgZOH19sgewMkHbyeN6IoxSM/AN5Epqn7CZQBkddQYhzImluG49p1a0u30WBxa0V8UW0LW6NNsaKf8JCoECRCa/6fWlT3B0bDGybhpBxQqeWBfs3I0khYqKP/InzHlaywDYFARAEm+RZkU2DLY/utkGjm1yOa4ZXkirb4= 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=fQO0p6hQ; arc=none smtp.client-ip=209.85.221.47 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="fQO0p6hQ" Received: by mail-wr1-f47.google.com with SMTP id ffacd0b85a97d-43b4d73463dso946284f8f.3 for ; Thu, 19 Mar 2026 13:25:36 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1773951935; x=1774556735; 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=+ZXgxsCauWvKSz1NA+qHkAzZKiqBTe7iEdmrrC0fzSs=; b=fQO0p6hQ5wPIe4cnsobl2IPAVkwgAX7lXtRU3KB2m9FtPqRn7jFOF1De0S4eU9zFeu NJ4bCzwTFeHPxDmqgF6p2/f5Hwz/5VawxP2Zz/T4PVZTBghTUhMv4a4pZvssgE6m/aRD hfsx0Rgb3BFSvo0ziv897cDV+HE49QU4iRCwoa1wKdXY46BYc5MUtVDEAD10T14OKKtr G8L936j2XL1c1ukMKUXcJlYloTSjZcB1fDNscpx0p79uVxkbW2BLa8zlk6ORYR23aLBt nteJ2ItzqDJPyrg1z4kaBOl0lg8I5kQ2CviRb3bdPgfpgrcmlXvQQb4q0KMM+n3sEAIw oumw== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20251104; t=1773951935; x=1774556735; 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=+ZXgxsCauWvKSz1NA+qHkAzZKiqBTe7iEdmrrC0fzSs=; b=mZq+sXfdUzrT27lRFR7t9e/dYHtlpiXQhgDU50m21tHmeUzTf5GLa8DLxu0luASUfr v2q4Q2rX8fP++Pct80gp6cEjsP7JnQVgsvAZWUNMTp5eaWJeM8syxMv9JgfulF5Cvo06 u1vdtwA1oBUVO7YMs79SMNdS0pFUKLl5ABuDghe6WMgrk6vapuJvqJv6ensyq0HsvpVE TaGvUfxsVFGgC5xjo30bv4ShqPS+tY1pM15kmMjWU/RNijVHGZsk4i77QFZtSeZf3vBi DjRj80nGGfVIQq7czJ9XXUxbI90yWLrj2G2sxiTJmfy4Uz540E95iVvCbFEJE6EakEIF yUcA== X-Gm-Message-State: AOJu0YyVxjoqmRLNkxlp6ySjhrMW5whryABCj/CFfsu+U0nrnpsliAo0 HEoI43AY8JAzp3i9/ZghDPaXrhsNGxt+RnBqVrbIr/sd1W+5wd6eqefeIDihONQBceQ= X-Gm-Gg: ATEYQzzHhBOI01n4kcsoS5ByE3feZkh1z4wAr5rM8S5ALLHEDIj6O19HTYhfjaLvi1p DQ4PpsSvjZHC2poFyuh6UUgDL5PI7d9i0HncLg93siQudeo6I7phlr1KMMcN/Nq1niCvNG8n95m mgE3sDVN6+qtfzqd0RP4JA1CJsgg7L7J3SjQma9LQS84s399VLFXLk0RFvAnyfc7BwuuI52yp2Y mWdTjxLlGuDHPulG6RvheLtK6m4I9wSAP5VbRi55b53Z7RD+15A2pcmMIxbToHGfZetClcs8NMj gJJ4xpf7GFT6IHmdXitcfxPuv5eeTPhMyEJmefndblzRY4d6G0qxE104aryTpG8jq/ycMZ/5Dzz mYRusa6laRIzowVUQl72+NNNYYOrT7vkDniEWUJ6TUXRaqH8g6+I4IFsQyjpBHyHs2oZ9/TIktR g9LRWyMCRX59b8TQOTzF3XjMVcuQuPdNvMlktll0Zay8qoqN/M X-Received: by 2002:a05:6000:2007:b0:43b:41df:705e with SMTP id ffacd0b85a97d-43b6428aa4fmr1299751f8f.49.1773951934625; Thu, 19 Mar 2026 13:25:34 -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.33 (version=TLS1_3 cipher=TLS_CHACHA20_POLY1305_SHA256 bits=256/256); Thu, 19 Mar 2026 13:25:34 -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 18/55] drivers: hv: dxgkrnl: Manage device allocation properties Date: Thu, 19 Mar 2026 20:24:32 +0000 Message-ID: <20260319202509.63802-19-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 ioctls to manage properties of a compute device allocation: - LX_DXUPDATEALLOCPROPERTY, - LX_DXSETALLOCATIONPRIORITY, - LX_DXGETALLOCATIONPRIORITY, - LX_DXQUERYALLOCATIONRESIDENCY. - LX_DXCHANGEVIDEOMEMORYRESERVATION, The LX_DXUPDATEALLOCPROPERTY ioctl requests the host to update various properties of a compute devoce allocation. The LX_DXSETALLOCATIONPRIORITY and LX_DXGETALLOCATIONPRIORITY ioctls are used to set/get allocation priority, which defines the importance of the allocation to be in the local device memory. The LX_DXQUERYALLOCATIONRESIDENCY ioctl queries if the allocation is located in the compute device accessible memory. The LX_DXCHANGEVIDEOMEMORYRESERVATION ioctl changes compute device memory reservation of an allocation. 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 | 21 +++ drivers/hv/dxgkrnl/dxgvmbus.c | 300 ++++++++++++++++++++++++++++++++++ drivers/hv/dxgkrnl/dxgvmbus.h | 50 ++++++ drivers/hv/dxgkrnl/ioctl.c | 217 +++++++++++++++++++++++- include/uapi/misc/d3dkmthk.h | 127 ++++++++++++++ 5 files changed, 708 insertions(+), 7 deletions(-) diff --git a/drivers/hv/dxgkrnl/dxgkrnl.h b/drivers/hv/dxgkrnl/dxgkrnl.h index 1d6b552f1c1a..7fefe4617488 100644 --- a/drivers/hv/dxgkrnl/dxgkrnl.h +++ b/drivers/hv/dxgkrnl/dxgkrnl.h @@ -851,6 +851,23 @@ int dxgvmb_send_lock2(struct dxgprocess *process, int dxgvmb_send_unlock2(struct dxgprocess *process, struct dxgadapter *adapter, struct d3dkmt_unlock2 *args); +int dxgvmb_send_update_alloc_property(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dddi_updateallocproperty *args, + struct d3dddi_updateallocproperty *__user + inargs); +int dxgvmb_send_set_allocation_priority(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dkmt_setallocationpriority *a); +int dxgvmb_send_get_allocation_priority(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dkmt_getallocationpriority *a); +int dxgvmb_send_change_vidmem_reservation(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dkmthandle other_process, + struct + d3dkmt_changevideomemoryreservation + *args); int dxgvmb_send_create_hwqueue(struct dxgprocess *process, struct dxgadapter *adapter, struct d3dkmt_createhwqueue *args, @@ -870,6 +887,10 @@ int dxgvmb_send_open_sync_object_nt(struct dxgprocess = *process, struct d3dkmt_opensyncobjectfromnthandle2 *args, struct dxgsyncobject *syncobj); +int dxgvmb_send_query_alloc_residency(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dkmt_queryallocationresidency + *args); 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 a80f84d9065a..dd2c97fee27b 100644 --- a/drivers/hv/dxgkrnl/dxgvmbus.c +++ b/drivers/hv/dxgkrnl/dxgvmbus.c @@ -1829,6 +1829,79 @@ int dxgvmb_send_destroy_allocation(struct dxgprocess= *process, return ret; } =20 +int dxgvmb_send_query_alloc_residency(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dkmt_queryallocationresidency + *args) +{ + int ret =3D -EINVAL; + struct dxgkvmb_command_queryallocationresidency *command =3D NULL; + u32 cmd_size =3D sizeof(*command); + u32 alloc_size =3D 0; + u32 result_allocation_size =3D 0; + struct dxgkvmb_command_queryallocationresidency_return *result =3D NULL; + u32 result_size =3D sizeof(*result); + struct dxgvmbusmsgres msg =3D {.hdr =3D NULL}; + + if (args->allocation_count > DXG_MAX_VM_BUS_PACKET_SIZE) { + ret =3D -EINVAL; + goto cleanup; + } + + if (args->allocation_count) { + alloc_size =3D args->allocation_count * + sizeof(struct d3dkmthandle); + cmd_size +=3D alloc_size; + result_allocation_size =3D args->allocation_count * + sizeof(args->residency_status[0]); + } else { + result_allocation_size =3D sizeof(args->residency_status[0]); + } + result_size +=3D result_allocation_size; + + ret =3D init_message_res(&msg, adapter, process, cmd_size, result_size); + if (ret) + goto cleanup; + command =3D (void *)msg.msg; + result =3D msg.res; + + command_vgpu_to_host_init2(&command->hdr, + DXGK_VMBCOMMAND_QUERYALLOCATIONRESIDENCY, + process->host_handle); + command->args =3D *args; + if (alloc_size) { + ret =3D copy_from_user(&command[1], args->allocations, + alloc_size); + if (ret) { + DXG_ERR("failed to copy alloc handles"); + ret =3D -EINVAL; + goto cleanup; + } + } + + ret =3D dxgvmb_send_sync_msg(msg.channel, msg.hdr, msg.size, + result, msg.res_size); + if (ret < 0) + goto cleanup; + + ret =3D ntstatus2int(result->status); + if (ret < 0) + goto cleanup; + + ret =3D copy_to_user(args->residency_status, &result[1], + result_allocation_size); + if (ret) { + DXG_ERR("failed to copy residency status"); + ret =3D -EINVAL; + } + +cleanup: + free_message((struct dxgvmbusmsg *)&msg, process); + if (ret) + DXG_TRACE("err: %d", ret); + return ret; +} + int dxgvmb_send_get_device_state(struct dxgprocess *process, struct dxgadapter *adapter, struct d3dkmt_getdevicestate *args, @@ -2461,6 +2534,233 @@ int dxgvmb_send_unlock2(struct dxgprocess *process, return ret; } =20 +int dxgvmb_send_update_alloc_property(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dddi_updateallocproperty *args, + struct d3dddi_updateallocproperty *__user + inargs) +{ + int ret; + int ret1; + struct dxgkvmb_command_updateallocationproperty *command; + struct dxgkvmb_command_updateallocationproperty_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_UPDATEALLOCATIONPROPERTY, + process->host_handle); + command->args =3D *args; + + ret =3D dxgvmb_send_sync_msg(msg.channel, msg.hdr, msg.size, + &result, sizeof(result)); + + if (ret < 0) + goto cleanup; + ret =3D ntstatus2int(result.status); + /* STATUS_PENING is a success code > 0 */ + if (ret =3D=3D STATUS_PENDING) { + ret1 =3D copy_to_user(&inargs->paging_fence_value, + &result.paging_fence_value, + sizeof(u64)); + if (ret1) { + DXG_ERR("failed to copy paging fence"); + ret =3D -EINVAL; + } + } +cleanup: + free_message(&msg, process); + if (ret) + DXG_TRACE("err: %d", ret); + return ret; +} + +int dxgvmb_send_set_allocation_priority(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dkmt_setallocationpriority *args) +{ + u32 cmd_size =3D sizeof(struct dxgkvmb_command_setallocationpriority); + u32 alloc_size =3D 0; + u32 priority_size =3D 0; + struct dxgkvmb_command_setallocationpriority *command; + int ret; + struct d3dkmthandle *allocations; + struct dxgvmbusmsg msg =3D {.hdr =3D NULL}; + + if (args->allocation_count > DXG_MAX_VM_BUS_PACKET_SIZE) { + ret =3D -EINVAL; + goto cleanup; + } + if (args->resource.v) { + priority_size =3D sizeof(u32); + if (args->allocation_count !=3D 0) { + ret =3D -EINVAL; + goto cleanup; + } + } else { + if (args->allocation_count =3D=3D 0) { + ret =3D -EINVAL; + goto cleanup; + } + alloc_size =3D args->allocation_count * + sizeof(struct d3dkmthandle); + cmd_size +=3D alloc_size; + priority_size =3D sizeof(u32) * args->allocation_count; + } + cmd_size +=3D priority_size; + + ret =3D init_message(&msg, adapter, process, cmd_size); + if (ret) + goto cleanup; + command =3D (void *)msg.msg; + + command_vgpu_to_host_init2(&command->hdr, + DXGK_VMBCOMMAND_SETALLOCATIONPRIORITY, + process->host_handle); + command->device =3D args->device; + command->allocation_count =3D args->allocation_count; + command->resource =3D args->resource; + allocations =3D (struct d3dkmthandle *) &command[1]; + ret =3D copy_from_user(allocations, args->allocation_list, + alloc_size); + if (ret) { + DXG_ERR("failed to copy alloc handle"); + ret =3D -EINVAL; + goto cleanup; + } + ret =3D copy_from_user((u8 *) allocations + alloc_size, + args->priorities, priority_size); + if (ret) { + DXG_ERR("failed to copy alloc priority"); + ret =3D -EINVAL; + goto cleanup; + } + + ret =3D dxgvmb_send_sync_msg_ntstatus(msg.channel, msg.hdr, msg.size); + +cleanup: + free_message(&msg, process); + if (ret) + DXG_TRACE("err: %d", ret); + return ret; +} + +int dxgvmb_send_get_allocation_priority(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dkmt_getallocationpriority *args) +{ + u32 cmd_size =3D sizeof(struct dxgkvmb_command_getallocationpriority); + u32 result_size; + u32 alloc_size =3D 0; + u32 priority_size =3D 0; + struct dxgkvmb_command_getallocationpriority *command; + struct dxgkvmb_command_getallocationpriority_return *result; + int ret; + struct d3dkmthandle *allocations; + struct dxgvmbusmsgres msg =3D {.hdr =3D NULL}; + + if (args->allocation_count > DXG_MAX_VM_BUS_PACKET_SIZE) { + ret =3D -EINVAL; + goto cleanup; + } + if (args->resource.v) { + priority_size =3D sizeof(u32); + if (args->allocation_count !=3D 0) { + ret =3D -EINVAL; + goto cleanup; + } + } else { + if (args->allocation_count =3D=3D 0) { + ret =3D -EINVAL; + goto cleanup; + } + alloc_size =3D args->allocation_count * + sizeof(struct d3dkmthandle); + cmd_size +=3D alloc_size; + priority_size =3D sizeof(u32) * args->allocation_count; + } + result_size =3D sizeof(*result) + priority_size; + + ret =3D init_message_res(&msg, adapter, process, cmd_size, result_size); + if (ret) + goto cleanup; + command =3D (void *)msg.msg; + result =3D msg.res; + + command_vgpu_to_host_init2(&command->hdr, + DXGK_VMBCOMMAND_GETALLOCATIONPRIORITY, + process->host_handle); + command->device =3D args->device; + command->allocation_count =3D args->allocation_count; + command->resource =3D args->resource; + allocations =3D (struct d3dkmthandle *) &command[1]; + ret =3D copy_from_user(allocations, args->allocation_list, + alloc_size); + if (ret) { + DXG_ERR("failed to copy alloc handles"); + ret =3D -EINVAL; + goto cleanup; + } + + ret =3D dxgvmb_send_sync_msg(msg.channel, msg.hdr, + msg.size + msg.res_size, + result, msg.res_size); + if (ret < 0) + goto cleanup; + + ret =3D ntstatus2int(result->status); + if (ret < 0) + goto cleanup; + + ret =3D copy_to_user(args->priorities, + (u8 *) result + sizeof(*result), + priority_size); + if (ret) { + DXG_ERR("failed to copy priorities"); + ret =3D -EINVAL; + } + +cleanup: + free_message((struct dxgvmbusmsg *)&msg, process); + if (ret) + DXG_TRACE("err: %d", ret); + return ret; +} + +int dxgvmb_send_change_vidmem_reservation(struct dxgprocess *process, + struct dxgadapter *adapter, + struct d3dkmthandle other_process, + struct + d3dkmt_changevideomemoryreservation + *args) +{ + struct dxgkvmb_command_changevideomemoryreservation *command; + int ret; + 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_CHANGEVIDEOMEMORYRESERVATION, + process->host_handle); + command->args =3D *args; + command->args.process =3D other_process.v; + + ret =3D dxgvmb_send_sync_msg_ntstatus(msg.channel, msg.hdr, msg.size); +cleanup: + free_message(&msg, process); + if (ret) + DXG_TRACE("err: %d", ret); + return ret; +} + int dxgvmb_send_create_hwqueue(struct dxgprocess *process, struct dxgadapter *adapter, struct d3dkmt_createhwqueue *args, diff --git a/drivers/hv/dxgkrnl/dxgvmbus.h b/drivers/hv/dxgkrnl/dxgvmbus.h index 447bb1ba391b..dbb01b9ab066 100644 --- a/drivers/hv/dxgkrnl/dxgvmbus.h +++ b/drivers/hv/dxgkrnl/dxgvmbus.h @@ -308,6 +308,29 @@ struct dxgkvmb_command_queryadapterinfo_return { u8 private_data[1]; }; =20 +/* Returns ntstatus */ +struct dxgkvmb_command_setallocationpriority { + struct dxgkvmb_command_vgpu_to_host hdr; + struct d3dkmthandle device; + struct d3dkmthandle resource; + u32 allocation_count; + /* struct d3dkmthandle allocations[allocation_count or 0]; */ + /* u32 priorities[allocation_count or 1]; */ +}; + +struct dxgkvmb_command_getallocationpriority { + struct dxgkvmb_command_vgpu_to_host hdr; + struct d3dkmthandle device; + struct d3dkmthandle resource; + u32 allocation_count; + /* struct d3dkmthandle allocations[allocation_count or 0]; */ +}; + +struct dxgkvmb_command_getallocationpriority_return { + struct ntstatus status; + /* u32 priorities[allocation_count or 1]; */ +}; + struct dxgkvmb_command_createdevice { struct dxgkvmb_command_vgpu_to_host hdr; struct d3dkmt_createdeviceflags flags; @@ -589,6 +612,22 @@ struct dxgkvmb_command_unlock2 { bool use_legacy_unlock; }; =20 +struct dxgkvmb_command_updateallocationproperty { + struct dxgkvmb_command_vgpu_to_host hdr; + struct d3dddi_updateallocproperty args; +}; + +struct dxgkvmb_command_updateallocationproperty_return { + u64 paging_fence_value; + struct ntstatus status; +}; + +/* Returns ntstatus */ +struct dxgkvmb_command_changevideomemoryreservation { + struct dxgkvmb_command_vgpu_to_host hdr; + struct d3dkmt_changevideomemoryreservation args; +}; + /* Returns the same structure */ struct dxgkvmb_command_createhwqueue { struct dxgkvmb_command_vgpu_to_host hdr; @@ -609,6 +648,17 @@ struct dxgkvmb_command_destroyhwqueue { struct d3dkmthandle hwqueue; }; =20 +struct dxgkvmb_command_queryallocationresidency { + struct dxgkvmb_command_vgpu_to_host hdr; + struct d3dkmt_queryallocationresidency args; + /* struct d3dkmthandle allocations[0 or number of allocations] */ +}; + +struct dxgkvmb_command_queryallocationresidency_return { + struct ntstatus status; + /* d3dkmt_allocationresidencystatus[NumAllocations] */ +}; + 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 37e218443310..b626e2518ff2 100644 --- a/drivers/hv/dxgkrnl/ioctl.c +++ b/drivers/hv/dxgkrnl/ioctl.c @@ -3214,7 +3214,7 @@ dxgkio_lock2(struct dxgprocess *process, void *__user= inargs) kref_put(&device->device_kref, dxgdevice_release); =20 success: - DXG_TRACE("ioctl:%s %s %d", errorstr(ret), __func__, ret); + DXG_TRACE("ioctl:%s %d", errorstr(ret), ret); return ret; } =20 @@ -3294,7 +3294,209 @@ dxgkio_unlock2(struct dxgprocess *process, void *__= user inargs) kref_put(&device->device_kref, dxgdevice_release); =20 success: - DXG_TRACE("ioctl:%s %s %d", errorstr(ret), __func__, ret); + DXG_TRACE("ioctl:%s %d", errorstr(ret), ret); + return ret; +} + +static int +dxgkio_update_alloc_property(struct dxgprocess *process, void *__user inar= gs) +{ + struct d3dddi_updateallocproperty args; + int ret; + struct dxgadapter *adapter =3D NULL; + struct dxgdevice *device =3D NULL; + + ret =3D copy_from_user(&args, inargs, sizeof(args)); + if (ret) { + DXG_ERR("failed to copy input args"); + ret =3D -EINVAL; + goto cleanup; + } + + device =3D dxgprocess_device_by_object_handle(process, + HMGRENTRY_TYPE_DXGPAGINGQUEUE, + args.paging_queue); + if (device =3D=3D NULL) { + ret =3D -EINVAL; + goto cleanup; + } + adapter =3D device->adapter; + ret =3D dxgadapter_acquire_lock_shared(adapter); + if (ret < 0) { + adapter =3D NULL; + goto cleanup; + } + + ret =3D dxgvmb_send_update_alloc_property(process, adapter, + &args, inargs); + +cleanup: + if (adapter) + dxgadapter_release_lock_shared(adapter); + + if (device) + kref_put(&device->device_kref, dxgdevice_release); + + DXG_TRACE("ioctl:%s %d", errorstr(ret), ret); + return ret; +} + +static int +dxgkio_query_alloc_residency(struct dxgprocess *process, void *__user inar= gs) +{ + struct d3dkmt_queryallocationresidency args; + struct dxgadapter *adapter =3D NULL; + struct dxgdevice *device =3D NULL; + int ret; + + 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.allocation_count =3D=3D 0) =3D=3D (args.resource.v =3D=3D 0)) { + ret =3D -EINVAL; + goto cleanup; + } + + device =3D dxgprocess_device_by_handle(process, args.device); + if (device =3D=3D NULL) { + ret =3D -EINVAL; + goto cleanup; + } + adapter =3D device->adapter; + ret =3D dxgadapter_acquire_lock_shared(adapter); + if (ret < 0) { + adapter =3D NULL; + goto cleanup; + } + ret =3D dxgvmb_send_query_alloc_residency(process, adapter, &args); +cleanup: + if (adapter) + dxgadapter_release_lock_shared(adapter); + if (device) + kref_put(&device->device_kref, dxgdevice_release); + DXG_TRACE("ioctl:%s %d", errorstr(ret), ret); + return ret; +} + +static int +dxgkio_set_allocation_priority(struct dxgprocess *process, void *__user in= args) +{ + struct d3dkmt_setallocationpriority args; + struct dxgadapter *adapter =3D NULL; + struct dxgdevice *device =3D NULL; + int ret; + + ret =3D copy_from_user(&args, inargs, sizeof(args)); + if (ret) { + DXG_ERR("failed to copy input args"); + ret =3D -EINVAL; + goto cleanup; + } + device =3D dxgprocess_device_by_handle(process, args.device); + if (device =3D=3D NULL) { + ret =3D -EINVAL; + goto cleanup; + } + adapter =3D device->adapter; + ret =3D dxgadapter_acquire_lock_shared(adapter); + if (ret < 0) { + adapter =3D NULL; + goto cleanup; + } + ret =3D dxgvmb_send_set_allocation_priority(process, adapter, &args); +cleanup: + if (adapter) + dxgadapter_release_lock_shared(adapter); + if (device) + kref_put(&device->device_kref, dxgdevice_release); + DXG_TRACE("ioctl:%s %d", errorstr(ret), ret); + return ret; +} + +static int +dxgkio_get_allocation_priority(struct dxgprocess *process, void *__user in= args) +{ + struct d3dkmt_getallocationpriority args; + struct dxgadapter *adapter =3D NULL; + struct dxgdevice *device =3D NULL; + int ret; + + ret =3D copy_from_user(&args, inargs, sizeof(args)); + if (ret) { + DXG_ERR("failed to copy input args"); + ret =3D -EINVAL; + goto cleanup; + } + device =3D dxgprocess_device_by_handle(process, args.device); + if (device =3D=3D NULL) { + ret =3D -EINVAL; + goto cleanup; + } + adapter =3D device->adapter; + ret =3D dxgadapter_acquire_lock_shared(adapter); + if (ret < 0) { + adapter =3D NULL; + goto cleanup; + } + ret =3D dxgvmb_send_get_allocation_priority(process, adapter, &args); +cleanup: + if (adapter) + dxgadapter_release_lock_shared(adapter); + if (device) + kref_put(&device->device_kref, dxgdevice_release); + DXG_TRACE("ioctl:%s %d", errorstr(ret), ret); + return ret; +} + +static int +dxgkio_change_vidmem_reservation(struct dxgprocess *process, void *__user = inargs) +{ + struct d3dkmt_changevideomemoryreservation 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("setting memory reservation for other 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.v =3D 0; + ret =3D dxgvmb_send_change_vidmem_reservation(process, adapter, + zerohandle, &args); + +cleanup: + + if (adapter_locked) + dxgadapter_release_lock_shared(adapter); + if (adapter) + kref_put(&adapter->adapter_kref, dxgadapter_release); + + DXG_TRACE("ioctl:%s %d", errorstr(ret), ret); return ret; } =20 @@ -4050,7 +4252,8 @@ static struct ioctl_desc ioctls[] =3D { /* 0x13 */ {dxgkio_destroy_allocation, LX_DXDESTROYALLOCATION2}, /* 0x14 */ {dxgkio_enum_adapters, LX_DXENUMADAPTERS2}, /* 0x15 */ {dxgkio_close_adapter, LX_DXCLOSEADAPTER}, -/* 0x16 */ {}, +/* 0x16 */ {dxgkio_change_vidmem_reservation, + LX_DXCHANGEVIDEOMEMORYRESERVATION}, /* 0x17 */ {}, /* 0x18 */ {dxgkio_create_hwqueue, LX_DXCREATEHWQUEUE}, /* 0x19 */ {dxgkio_destroy_device, LX_DXDESTROYDEVICE}, @@ -4070,11 +4273,11 @@ static struct ioctl_desc ioctls[] =3D { /* 0x27 */ {}, /* 0x28 */ {}, /* 0x29 */ {}, -/* 0x2a */ {}, +/* 0x2a */ {dxgkio_query_alloc_residency, LX_DXQUERYALLOCATIONRESIDENCY}, /* 0x2b */ {}, /* 0x2c */ {}, /* 0x2d */ {}, -/* 0x2e */ {}, +/* 0x2e */ {dxgkio_set_allocation_priority, LX_DXSETALLOCATIONPRIORITY}, /* 0x2f */ {}, /* 0x30 */ {}, /* 0x31 */ {dxgkio_signal_sync_object_cpu, @@ -4089,13 +4292,13 @@ static struct ioctl_desc ioctls[] =3D { /* 0x36 */ {dxgkio_submit_wait_to_hwqueue, LX_DXSUBMITWAITFORSYNCOBJECTSTOHWQUEUE}, /* 0x37 */ {dxgkio_unlock2, LX_DXUNLOCK2}, -/* 0x38 */ {}, +/* 0x38 */ {dxgkio_update_alloc_property, LX_DXUPDATEALLOCPROPERTY}, /* 0x39 */ {}, /* 0x3a */ {dxgkio_wait_sync_object_cpu, LX_DXWAITFORSYNCHRONIZATIONOBJECTFROMCPU}, /* 0x3b */ {dxgkio_wait_sync_object_gpu, LX_DXWAITFORSYNCHRONIZATIONOBJECTFROMGPU}, -/* 0x3c */ {}, +/* 0x3c */ {dxgkio_get_allocation_priority, LX_DXGETALLOCATIONPRIORITY}, /* 0x3d */ {}, /* 0x3e */ {dxgkio_enum_adapters3, LX_DXENUMADAPTERS3}, /* 0x3f */ {dxgkio_share_objects, LX_DXSHAREOBJECTS}, diff --git a/include/uapi/misc/d3dkmthk.h b/include/uapi/misc/d3dkmthk.h index b498f09e694d..af381101fd90 100644 --- a/include/uapi/misc/d3dkmthk.h +++ b/include/uapi/misc/d3dkmthk.h @@ -668,6 +668,63 @@ struct d3dkmt_submitcommandtohwqueue { #endif }; =20 +struct d3dkmt_setallocationpriority { + struct d3dkmthandle device; + struct d3dkmthandle resource; +#ifdef __KERNEL__ + const struct d3dkmthandle *allocation_list; +#else + __u64 allocation_list; +#endif + __u32 allocation_count; + __u32 reserved; +#ifdef __KERNEL__ + const __u32 *priorities; +#else + __u64 priorities; +#endif +}; + +struct d3dkmt_getallocationpriority { + struct d3dkmthandle device; + struct d3dkmthandle resource; +#ifdef __KERNEL__ + const struct d3dkmthandle *allocation_list; +#else + __u64 allocation_list; +#endif + __u32 allocation_count; + __u32 reserved; +#ifdef __KERNEL__ + __u32 *priorities; +#else + __u64 priorities; +#endif +}; + +enum d3dkmt_allocationresidencystatus { + _D3DKMT_ALLOCATIONRESIDENCYSTATUS_RESIDENTINGPUMEMORY =3D 1, + _D3DKMT_ALLOCATIONRESIDENCYSTATUS_RESIDENTINSHAREDMEMORY =3D 2, + _D3DKMT_ALLOCATIONRESIDENCYSTATUS_NOTRESIDENT =3D 3, +}; + +struct d3dkmt_queryallocationresidency { + struct d3dkmthandle device; + struct d3dkmthandle resource; +#ifdef __KERNEL__ + struct d3dkmthandle *allocations; +#else + __u64 allocations; +#endif + __u32 allocation_count; + __u32 reserved; +#ifdef __KERNEL__ + enum d3dkmt_allocationresidencystatus *residency_status; +#else + __u64 residency_status; +#endif +}; + struct d3dddicb_lock2flags { union { struct { @@ -835,6 +892,11 @@ struct d3dkmt_destroyallocation2 { struct d3dddicb_destroyallocation2flags flags; }; =20 +enum d3dkmt_memory_segment_group { + _D3DKMT_MEMORY_SEGMENT_GROUP_LOCAL =3D 0, + _D3DKMT_MEMORY_SEGMENT_GROUP_NON_LOCAL =3D 1 +}; + struct d3dkmt_adaptertype { union { struct { @@ -886,6 +948,61 @@ struct d3dddi_openallocationinfo2 { __u64 reserved[6]; }; =20 +struct d3dddi_updateallocproperty_flags { + union { + struct { + __u32 accessed_physically:1; + __u32 reserved:31; + }; + __u32 value; + }; +}; + +struct d3dddi_segmentpreference { + union { + struct { + __u32 segment_id0:5; + __u32 direction0:1; + __u32 segment_id1:5; + __u32 direction1:1; + __u32 segment_id2:5; + __u32 direction2:1; + __u32 segment_id3:5; + __u32 direction3:1; + __u32 segment_id4:5; + __u32 direction4:1; + __u32 reserved:2; + }; + __u32 value; + }; +}; + +struct d3dddi_updateallocproperty { + struct d3dkmthandle paging_queue; + struct d3dkmthandle allocation; + __u32 supported_segment_set; + struct d3dddi_segmentpreference preferred_segment; + struct d3dddi_updateallocproperty_flags flags; + __u64 paging_fence_value; + union { + struct { + __u32 set_accessed_physically:1; + __u32 set_supported_segmentSet:1; + __u32 set_preferred_segment:1; + __u32 reserved:29; + }; + __u32 property_mask_value; + }; +}; + +struct d3dkmt_changevideomemoryreservation { + __u64 process; + struct d3dkmthandle adapter; + enum d3dkmt_memory_segment_group memory_segment_group; + __u64 reservation; + __u32 physical_adapter_index; +}; + struct d3dkmt_createhwqueue { struct d3dkmthandle context; struct d3dddi_createhwqueueflags flags; @@ -1099,6 +1216,8 @@ struct d3dkmt_shareobjectwithhost { _IOWR(0x47, 0x14, struct d3dkmt_enumadapters2) #define LX_DXCLOSEADAPTER \ _IOWR(0x47, 0x15, struct d3dkmt_closeadapter) +#define LX_DXCHANGEVIDEOMEMORYRESERVATION \ + _IOWR(0x47, 0x16, struct d3dkmt_changevideomemoryreservation) #define LX_DXCREATEHWQUEUE \ _IOWR(0x47, 0x18, struct d3dkmt_createhwqueue) #define LX_DXDESTROYHWQUEUE \ @@ -1111,6 +1230,10 @@ struct d3dkmt_shareobjectwithhost { _IOWR(0x47, 0x1d, struct d3dkmt_destroysynchronizationobject) #define LX_DXLOCK2 \ _IOWR(0x47, 0x25, struct d3dkmt_lock2) +#define LX_DXQUERYALLOCATIONRESIDENCY \ + _IOWR(0x47, 0x2a, struct d3dkmt_queryallocationresidency) +#define LX_DXSETALLOCATIONPRIORITY \ + _IOWR(0x47, 0x2e, struct d3dkmt_setallocationpriority) #define LX_DXSIGNALSYNCHRONIZATIONOBJECTFROMCPU \ _IOWR(0x47, 0x31, struct d3dkmt_signalsynchronizationobjectfromcpu) #define LX_DXSIGNALSYNCHRONIZATIONOBJECTFROMGPU \ @@ -1125,10 +1248,14 @@ struct d3dkmt_shareobjectwithhost { _IOWR(0x47, 0x36, struct d3dkmt_submitwaitforsyncobjectstohwqueue) #define LX_DXUNLOCK2 \ _IOWR(0x47, 0x37, struct d3dkmt_unlock2) +#define LX_DXUPDATEALLOCPROPERTY \ + _IOWR(0x47, 0x38, struct d3dddi_updateallocproperty) #define LX_DXWAITFORSYNCHRONIZATIONOBJECTFROMCPU \ _IOWR(0x47, 0x3a, struct d3dkmt_waitforsynchronizationobjectfromcpu) #define LX_DXWAITFORSYNCHRONIZATIONOBJECTFROMGPU \ _IOWR(0x47, 0x3b, struct d3dkmt_waitforsynchronizationobjectfromgpu) +#define LX_DXGETALLOCATIONPRIORITY \ + _IOWR(0x47, 0x3c, struct d3dkmt_getallocationpriority) #define LX_DXENUMADAPTERS3 \ _IOWR(0x47, 0x3e, struct d3dkmt_enumadapters3) #define LX_DXSHAREOBJECTS \