From: Dmytro Laktyushkin <dmytro.laktyush...@amd.com>

Currently vmid is decided internally inside dc. With the introduction
of new asics we are required to coordinate vmid use with external
components.

This change converts vmid logic to a DAL module allowing vmid to be
passed in as a parameter to DC.

Signed-off-by: Dmytro Laktyushkin <dmytro.laktyush...@amd.com>
Reviewed-by: Charlene Liu <charlene....@amd.com>
Acked-by: Bhawanpreet Lakha <bhawanpreet.la...@amd.com>
Signed-off-by: Alex Deucher <alexander.deuc...@amd.com>
---
 drivers/gpu/drm/amd/display/dc/Makefile       |   7 +-
 drivers/gpu/drm/amd/display/dc/core/dc.c      |  20 ---
 .../drm/amd/display/dc/core/dc_vm_helper.c    |  93 +++--------
 drivers/gpu/drm/amd/display/dc/dc.h           |  17 +-
 drivers/gpu/drm/amd/display/dc/dc_hw_types.h  |   2 +
 .../gpu/drm/amd/display/dc/dcn10/dcn10_hubp.c |   3 +-
 .../gpu/drm/amd/display/dc/dcn10/dcn10_hubp.h |   3 +-
 .../amd/display/dc/dcn10/dcn10_hw_sequencer.c |   3 +-
 .../drm/amd/display/dc/dcn20/dcn20_hubbub.c   |  72 ++++----
 .../drm/amd/display/dc/dcn20/dcn20_hubbub.h   |  12 +-
 .../gpu/drm/amd/display/dc/dcn20/dcn20_hubp.c |   5 +-
 .../gpu/drm/amd/display/dc/dcn20/dcn20_hubp.h |   3 +-
 .../drm/amd/display/dc/dcn20/dcn20_hwseq.c    |  83 ++++-----
 .../drm/amd/display/dc/dcn20/dcn20_resource.c |   2 +-
 .../gpu/drm/amd/display/dc/dcn20/dcn20_vmid.c |   7 +-
 .../gpu/drm/amd/display/dc/dcn20/dcn20_vmid.h |   1 -
 .../gpu/drm/amd/display/dc/inc/hw/dchubbub.h  |  16 +-
 drivers/gpu/drm/amd/display/dc/inc/hw/hubp.h  |   3 +-
 drivers/gpu/drm/amd/display/dc/inc/hw/vmid.h  |   1 +
 .../gpu/drm/amd/display/dc/inc/hw_sequencer.h |  17 +-
 .../gpu/drm/amd/display/dc/inc/vm_helper.h    |  16 +-
 .../drm/amd/display/modules/inc/mod_vmid.h    |  44 +++++
 .../gpu/drm/amd/display/modules/vmid/vmid.c   | 158 ++++++++++++++++++
 23 files changed, 357 insertions(+), 231 deletions(-)
 create mode 100644 drivers/gpu/drm/amd/display/modules/inc/mod_vmid.h
 create mode 100644 drivers/gpu/drm/amd/display/modules/vmid/vmid.c

diff --git a/drivers/gpu/drm/amd/display/dc/Makefile 
b/drivers/gpu/drm/amd/display/dc/Makefile
index 00eaa69ba53d..55ce5b657390 100644
--- a/drivers/gpu/drm/amd/display/dc/Makefile
+++ b/drivers/gpu/drm/amd/display/dc/Makefile
@@ -50,8 +50,11 @@ AMD_DC = $(addsuffix /Makefile, $(addprefix 
$(FULL_AMD_DISPLAY_PATH)/dc/,$(DC_LI
 include $(AMD_DC)
 
 DISPLAY_CORE = dc.o dc_link.o dc_resource.o dc_hw_sequencer.o dc_sink.o \
-dc_surface.o dc_link_hwss.o dc_link_dp.o dc_link_ddc.o dc_debug.o dc_stream.o \
-dc_vm_helper.o
+dc_surface.o dc_link_hwss.o dc_link_dp.o dc_link_ddc.o dc_debug.o dc_stream.o
+
+ifdef CONFIG_DRM_AMD_DC_DCN2_0
+DISPLAY_CORE += dc_vm_helper.o
+endif
 
 AMD_DISPLAY_CORE = $(addprefix $(AMDDALPATH)/dc/core/,$(DISPLAY_CORE))
 
diff --git a/drivers/gpu/drm/amd/display/dc/core/dc.c 
b/drivers/gpu/drm/amd/display/dc/core/dc.c
index 8bc8ac3a7718..a29bb9bf3f9c 100644
--- a/drivers/gpu/drm/amd/display/dc/core/dc.c
+++ b/drivers/gpu/drm/amd/display/dc/core/dc.c
@@ -808,26 +808,6 @@ void dc_destroy(struct dc **dc)
        *dc = NULL;
 }
 
-#ifdef CONFIG_DRM_AMD_DC_DCN2_0
-bool dc_init_memory_hub(struct dc *dc, struct dc_addr_space_config *config)
-{
-       // Memory hub init isn't done as part of dc_create because in windows, 
dal/dc is
-       // constructed before the vm config is setup in kmd so there's no way
-       // they can give it to us at boot/dc_create
-       bool vmSupported;
-
-       // Call HWSS to setup HUBBUB for address config
-       dc->hwss.init_dchub(dc->hwseq, dc, config);
-
-       // Pre-init system aperture start/end for all HUBP instances (if not 
gating?)
-       // or cache system aperture if using power gating
-       memcpy(&dc->vm_config, config, sizeof(struct dc_addr_space_config));
-
-       vmSupported = (dc->ctx->asic_id.chip_family == FAMILY_NV) ? true : 
false;
-       return vmSupported;
-}
-
-#endif
 static void enable_timing_multisync(
                struct dc *dc,
                struct dc_state *ctx)
diff --git a/drivers/gpu/drm/amd/display/dc/core/dc_vm_helper.c 
b/drivers/gpu/drm/amd/display/dc/core/dc_vm_helper.c
index 6ce87b682a32..a96d8de9380e 100644
--- a/drivers/gpu/drm/amd/display/dc/core/dc_vm_helper.c
+++ b/drivers/gpu/drm/amd/display/dc/core/dc_vm_helper.c
@@ -24,8 +24,9 @@
  */
 
 #include "vm_helper.h"
+#include "dc.h"
 
-static void mark_vmid_used(struct vm_helper *vm_helper, unsigned int pos, 
uint8_t hubp_idx)
+void vm_helper_mark_vmid_used(struct vm_helper *vm_helper, unsigned int pos, 
uint8_t hubp_idx)
 {
        struct vmid_usage vmids = vm_helper->hubp_vmid_usage[hubp_idx];
 
@@ -33,91 +34,43 @@ static void mark_vmid_used(struct vm_helper *vm_helper, 
unsigned int pos, uint8_
        vmids.vmid_usage[1] = 1 << pos;
 }
 
-static void add_ptb_to_table(struct vm_helper *vm_helper, unsigned int vmid, 
uint64_t ptb)
+int dc_setup_system_context(struct dc *dc, struct dc_phy_addr_space_config 
*pa_config)
 {
-       vm_helper->ptb_assigned_to_vmid[vmid] = ptb;
-       vm_helper->num_vmids_available--;
-}
-
-static void clear_entry_from_vmid_table(struct vm_helper *vm_helper, unsigned 
int vmid)
-{
-       vm_helper->ptb_assigned_to_vmid[vmid] = 0;
-       vm_helper->num_vmids_available++;
-}
-
-static void evict_vmids(struct vm_helper *vm_helper)
-{
-       int i;
-       uint16_t ord = 0;
+       int num_vmids = 0;
 
-       for (i = 0; i < vm_helper->num_vmid; i++)
-               ord |= vm_helper->hubp_vmid_usage[i].vmid_usage[0] | 
vm_helper->hubp_vmid_usage[i].vmid_usage[1];
+       /* Call HWSS to setup HUBBUB for address config */
+       if (dc->hwss.init_sys_ctx) {
+               num_vmids = dc->hwss.init_sys_ctx(dc->hwseq, dc, pa_config);
 
-       // At this point any positions with value 0 are unused vmids, evict them
-       for (i = 1; i < vm_helper->num_vmid; i++) {
-               if (ord & (1u << i))
-                       clear_entry_from_vmid_table(vm_helper, i);
+               /* Pre-init system aperture start/end for all HUBP instances 
(if not gating?)
+                * or cache system aperture if using power gating
+                */
+               memcpy(&dc->vm_pa_config, pa_config, sizeof(struct 
dc_phy_addr_space_config));
+               dc->vm_pa_config.valid = true;
        }
-}
-
-// Return value of -1 indicates vmid table unitialized or ptb dne in the table
-static int get_existing_vmid_for_ptb(struct vm_helper *vm_helper, uint64_t ptb)
-{
-       int i;
 
-       for (i = 0; i < vm_helper->num_vmid; i++) {
-               if (vm_helper->ptb_assigned_to_vmid[i] == ptb)
-                       return i;
-       }
-
-       return -1;
+       return num_vmids;
 }
 
-// Expected to be called only when there's an available vmid
-static int get_next_available_vmid(struct vm_helper *vm_helper)
+void dc_setup_vm_context(struct dc *dc, struct dc_virtual_addr_space_config 
*va_config, int vmid)
 {
-       int i;
-
-       for (i = 1; i < vm_helper->num_vmid; i++) {
-               if (vm_helper->ptb_assigned_to_vmid[i] == 0)
-                       return i;
-       }
-
-       return -1;
+       dc->hwss.init_vm_ctx(dc->hwseq, dc, va_config, vmid);
 }
 
-uint8_t get_vmid_for_ptb(struct vm_helper *vm_helper, int64_t ptb, uint8_t 
hubp_idx)
+int dc_get_vmid_use_vector(struct dc *dc)
 {
-       unsigned int vmid = 0;
-       int vmid_exists = -1;
-
-       // Physical address gets vmid 0
-       if (ptb == 0)
-               return 0;
-
-       vmid_exists = get_existing_vmid_for_ptb(vm_helper, ptb);
-
-       if (vmid_exists != -1) {
-               mark_vmid_used(vm_helper, vmid_exists, hubp_idx);
-               vmid = vmid_exists;
-       } else {
-               if (vm_helper->num_vmids_available == 0)
-                       evict_vmids(vm_helper);
-
-               vmid = get_next_available_vmid(vm_helper);
-               mark_vmid_used(vm_helper, vmid, hubp_idx);
-               add_ptb_to_table(vm_helper, vmid, ptb);
-       }
+       int i;
+       int in_use = 0;
 
-       return vmid;
+       for (i = 0; i < dc->vm_helper->num_vmid; i++)
+               in_use |= dc->vm_helper->hubp_vmid_usage[i].vmid_usage[0]
+                       | dc->vm_helper->hubp_vmid_usage[i].vmid_usage[1];
+       return in_use;
 }
 
-void init_vm_helper(struct vm_helper *vm_helper, unsigned int num_vmid, 
unsigned int num_hubp)
+void vm_helper_init(struct vm_helper *vm_helper, unsigned int num_vmid)
 {
        vm_helper->num_vmid = num_vmid;
-       vm_helper->num_hubp = num_hubp;
-       vm_helper->num_vmids_available = num_vmid - 1;
 
        memset(vm_helper->hubp_vmid_usage, 0, 
sizeof(vm_helper->hubp_vmid_usage[0]) * MAX_HUBP);
-       memset(vm_helper->ptb_assigned_to_vmid, 0, 
sizeof(vm_helper->ptb_assigned_to_vmid[0]) * MAX_VMID);
 }
diff --git a/drivers/gpu/drm/amd/display/dc/dc.h 
b/drivers/gpu/drm/amd/display/dc/dc.h
index 07a264fdce4e..5ebfeee9d08e 100644
--- a/drivers/gpu/drm/amd/display/dc/dc.h
+++ b/drivers/gpu/drm/amd/display/dc/dc.h
@@ -403,21 +403,17 @@ struct dc_phy_addr_space_config {
                uint64_t page_table_end_addr;
                uint64_t page_table_base_addr;
        } gart_config;
+
+       bool valid;
 };
 
 struct dc_virtual_addr_space_config {
+       uint64_t        page_table_base_addr;
        uint64_t        page_table_start_addr;
        uint64_t        page_table_end_addr;
        uint32_t        page_table_block_size_in_bytes;
        uint8_t         page_table_depth; // 1 = 1 level, 2 = 2 level, etc.  0 
= invalid
 };
-
-struct dc_addr_space_config {
-       struct dc_phy_addr_space_config         pa_config;
-       struct dc_virtual_addr_space_config     va_config;
-       uint32_t        valid:1;
-};
-
 #endif
 
 struct dc_bounding_box_overrides {
@@ -449,7 +445,7 @@ struct dc {
 #endif
        struct dc_context *ctx;
 #ifdef CONFIG_DRM_AMD_DC_DCN2_0
-       struct dc_addr_space_config vm_config;
+       struct dc_phy_addr_space_config vm_pa_config;
 #endif
 
        uint8_t link_count;
@@ -539,8 +535,11 @@ struct dc_callback_init {
 };
 
 struct dc *dc_create(const struct dc_init_data *init_params);
+int dc_get_vmid_use_vector(struct dc *dc);
 #ifdef CONFIG_DRM_AMD_DC_DCN2_0
-bool dc_init_memory_hub(struct dc *dc, struct dc_addr_space_config *config);
+void dc_setup_vm_context(struct dc *dc, struct dc_virtual_addr_space_config 
*va_config, int vmid);
+/* Returns the number of vmids supported */
+int dc_setup_system_context(struct dc *dc, struct dc_phy_addr_space_config 
*pa_config);
 #endif
 void dc_init_callbacks(struct dc *dc,
                const struct dc_callback_init *init_params);
diff --git a/drivers/gpu/drm/amd/display/dc/dc_hw_types.h 
b/drivers/gpu/drm/amd/display/dc/dc_hw_types.h
index 6b1431d63dab..22db5682aa6c 100644
--- a/drivers/gpu/drm/amd/display/dc/dc_hw_types.h
+++ b/drivers/gpu/drm/amd/display/dc/dc_hw_types.h
@@ -99,6 +99,8 @@ struct dc_plane_address {
        };
 
        union large_integer page_table_base;
+
+       uint8_t vmid;
 };
 
 struct dc_size {
diff --git a/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hubp.c 
b/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hubp.c
index a48d314011a9..934bacc0c6ad 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hubp.c
+++ b/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hubp.c
@@ -339,8 +339,7 @@ void hubp1_program_pixel_format(
 bool hubp1_program_surface_flip_and_addr(
        struct hubp *hubp,
        const struct dc_plane_address *address,
-       bool flip_immediate,
-       uint8_t vmid)
+       bool flip_immediate)
 {
        struct dcn10_hubp *hubp1 = TO_DCN10_HUBP(hubp);
 
diff --git a/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hubp.h 
b/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hubp.h
index 91dc42e580ec..31c8fdd3206c 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hubp.h
+++ b/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hubp.h
@@ -720,8 +720,7 @@ void hubp1_dcc_control(struct hubp *hubp,
 bool hubp1_program_surface_flip_and_addr(
        struct hubp *hubp,
        const struct dc_plane_address *address,
-       bool flip_immediate,
-       uint8_t vmid);
+       bool flip_immediate);
 
 #endif
 bool hubp1_is_flip_pending(struct hubp *hubp);
diff --git a/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hw_sequencer.c 
b/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hw_sequencer.c
index 0b1503de0f7a..e6b7e10c612f 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hw_sequencer.c
+++ b/drivers/gpu/drm/amd/display/dc/dcn10/dcn10_hw_sequencer.c
@@ -1296,8 +1296,7 @@ static void dcn10_update_plane_addr(const struct dc *dc, 
struct pipe_ctx *pipe_c
        pipe_ctx->plane_res.hubp->funcs->hubp_program_surface_flip_and_addr(
                        pipe_ctx->plane_res.hubp,
                        &plane_state->address,
-                       plane_state->flip_immediate,
-                       0);
+                       plane_state->flip_immediate);
 
        plane_state->status.requested_address = plane_state->address;
 
diff --git a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubbub.c 
b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubbub.c
index 724f1c5ef614..ece6e136437b 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubbub.c
+++ b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubbub.c
@@ -47,6 +47,11 @@
 #define FN(reg_name, field_name) \
        hubbub1->shifts->field_name, hubbub1->masks->field_name
 
+#ifdef NUM_VMID
+#undef NUM_VMID
+#endif
+#define NUM_VMID 16
+
 bool hubbub2_dcc_support_swizzle(
                enum swizzle_mode_values swizzle,
                unsigned int bytes_per_element,
@@ -294,15 +299,6 @@ bool hubbub2_get_dcc_compression_cap(struct hubbub *hubbub,
        return true;
 }
 
-void hubbub2_setup_vmid_ptb(struct hubbub *hubbub,
-               uint64_t ptb,
-               uint8_t vmid)
-{
-       struct dcn20_hubbub *hubbub1 = TO_DCN20_HUBBUB(hubbub);
-
-       dcn20_vmid_set_ptb(&hubbub1->vmid[vmid], ptb);
-}
-
 static enum dcn_hubbub_page_table_depth page_table_depth_to_hw(unsigned int 
page_table_depth)
 {
        enum dcn_hubbub_page_table_depth depth = 0;
@@ -347,49 +343,53 @@ static enum dcn_hubbub_page_table_block_size 
page_table_block_size_to_hw(unsigne
        return block_size;
 }
 
-void hubbub2_init_dchub(struct hubbub *hubbub,
-               struct hubbub_addr_config *config)
+void hubbub2_init_vm_ctx(struct hubbub *hubbub,
+               struct dcn_hubbub_virt_addr_config *va_config,
+               int vmid)
 {
-       int i;
        struct dcn20_hubbub *hubbub1 = TO_DCN20_HUBBUB(hubbub);
-       struct dcn_vmid_page_table_config phys_config;
        struct dcn_vmid_page_table_config virt_config;
 
+       virt_config.page_table_start_addr = va_config->page_table_start_addr >> 
12;
+       virt_config.page_table_end_addr = va_config->page_table_end_addr >> 12;
+       virt_config.depth = page_table_depth_to_hw(va_config->page_table_depth);
+       virt_config.block_size = 
page_table_block_size_to_hw(va_config->page_table_block_size);
+       virt_config.page_table_base_addr = va_config->page_table_base_addr;
+
+       dcn20_vmid_setup(&hubbub1->vmid[vmid], &virt_config);
+}
+
+int hubbub2_init_dchub_sys_ctx(struct hubbub *hubbub,
+               struct dcn_hubbub_phys_addr_config *pa_config)
+{
+       struct dcn20_hubbub *hubbub1 = TO_DCN20_HUBBUB(hubbub);
+       struct dcn_vmid_page_table_config phys_config;
+
        REG_SET(DCN_VM_FB_LOCATION_BASE, 0,
-                       FB_BASE, config->pa_config.system_aperture.fb_base);
+                       FB_BASE, pa_config->system_aperture.fb_base);
        REG_SET(DCN_VM_FB_LOCATION_TOP, 0,
-                       FB_TOP, config->pa_config.system_aperture.fb_top);
+                       FB_TOP, pa_config->system_aperture.fb_top);
        REG_SET(DCN_VM_FB_OFFSET, 0,
-                       FB_OFFSET, config->pa_config.system_aperture.fb_offset);
+                       FB_OFFSET, pa_config->system_aperture.fb_offset);
        REG_SET(DCN_VM_AGP_BOT, 0,
-                       AGP_BOT, config->pa_config.system_aperture.agp_bot);
+                       AGP_BOT, pa_config->system_aperture.agp_bot);
        REG_SET(DCN_VM_AGP_TOP, 0,
-                       AGP_TOP, config->pa_config.system_aperture.agp_top);
+                       AGP_TOP, pa_config->system_aperture.agp_top);
        REG_SET(DCN_VM_AGP_BASE, 0,
-                       AGP_BASE, config->pa_config.system_aperture.agp_base);
+                       AGP_BASE, pa_config->system_aperture.agp_base);
 
-       if (config->pa_config.gart_config.page_table_start_addr != 
config->pa_config.gart_config.page_table_end_addr) {
+       if (pa_config->gart_config.page_table_start_addr != 
pa_config->gart_config.page_table_end_addr) {
                phys_config.depth = 1;
                phys_config.block_size = 4096;
-               phys_config.page_table_start_addr = 
config->pa_config.gart_config.page_table_start_addr >> 12;
-               phys_config.page_table_end_addr = 
config->pa_config.gart_config.page_table_end_addr >> 12;
+               phys_config.page_table_start_addr = 
pa_config->gart_config.page_table_start_addr >> 12;
+               phys_config.page_table_end_addr = 
pa_config->gart_config.page_table_end_addr >> 12;
+               phys_config.page_table_base_addr = 
pa_config->gart_config.page_table_base_addr;
 
                // Init VMID 0 based on PA config
                dcn20_vmid_setup(&hubbub1->vmid[0], &phys_config);
-               dcn20_vmid_set_ptb(&hubbub1->vmid[0], 
config->pa_config.gart_config.page_table_base_addr);
        }
 
-       if (config->va_config.page_table_start_addr != 
config->va_config.page_table_end_addr) {
-               // Init VMID 1-15 based on VA config
-               for (i = 1; i < 16; i++) {
-                       virt_config.page_table_start_addr = 
config->va_config.page_table_start_addr >> 12;
-                       virt_config.page_table_end_addr = 
config->va_config.page_table_end_addr >> 12;
-                       virt_config.depth = 
page_table_depth_to_hw(config->va_config.page_table_depth);
-                       virt_config.block_size = 
page_table_block_size_to_hw(config->va_config.page_table_block_size);
-
-                       dcn20_vmid_setup(&hubbub1->vmid[i], &virt_config);
-               }
-       }
+       return NUM_VMID;
 }
 
 void hubbub2_update_dchub(struct hubbub *hubbub,
@@ -564,8 +564,8 @@ static void hubbub2_program_watermarks(
 
 static const struct hubbub_funcs hubbub2_funcs = {
        .update_dchub = hubbub2_update_dchub,
-       .init_dchub = hubbub2_init_dchub,
-       .setup_vmid_ptb = hubbub2_setup_vmid_ptb,
+       .init_dchub_sys_ctx = hubbub2_init_dchub_sys_ctx,
+       .init_vm_ctx = hubbub2_init_vm_ctx,
        .dcc_support_swizzle = hubbub2_dcc_support_swizzle,
        .dcc_support_pixel_format = hubbub2_dcc_support_pixel_format,
        .get_dcc_compression_cap = hubbub2_get_dcc_compression_cap,
diff --git a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubbub.h 
b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubbub.h
index 6ca4a5a9f8c7..a7b6ca26a9ad 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubbub.h
+++ b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubbub.h
@@ -89,9 +89,11 @@ bool hubbub2_initialize_vmids(struct hubbub *hubbub,
                const struct dc_dcc_surface_param *input,
                struct dc_surface_dcc_cap *output);
 
-void hubbub2_init_dchub(struct hubbub *hubbub,
-               struct hubbub_addr_config *config);
-
+int hubbub2_init_dchub_sys_ctx(struct hubbub *hubbub,
+               struct dcn_hubbub_phys_addr_config *pa_config);
+void hubbub2_init_vm_ctx(struct hubbub *hubbub,
+               struct dcn_hubbub_virt_addr_config *va_config,
+               int vmid);
 void hubbub2_update_dchub(struct hubbub *hubbub,
                struct dchub_init_data *dh_data);
 
@@ -102,8 +104,4 @@ void hubbub2_get_dchub_ref_freq(struct hubbub *hubbub,
 void hubbub2_wm_read_state(struct hubbub *hubbub,
                struct dcn_hubbub_wm *wm);
 
-void hubbub2_setup_vmid_ptb(struct hubbub *hubbub,
-               uint64_t ptb,
-               uint8_t vmid);
-
 #endif
diff --git a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubp.c 
b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubp.c
index dcba2c5326b2..d3f7dd374d50 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubp.c
+++ b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubp.c
@@ -447,8 +447,7 @@ bool hubp2_dmdata_status_done(struct hubp *hubp)
 bool hubp2_program_surface_flip_and_addr(
        struct hubp *hubp,
        const struct dc_plane_address *address,
-       bool flip_immediate,
-       uint8_t vmid)
+       bool flip_immediate)
 {
        struct dcn20_hubp *hubp2 = TO_DCN20_HUBP(hubp);
 
@@ -458,7 +457,7 @@ bool hubp2_program_surface_flip_and_addr(
 
        // Program VMID reg
        REG_UPDATE(VMID_SETTINGS_0,
-                       VMID, vmid);
+                       VMID, address->vmid);
 
        if (address->type == PLN_ADDR_TYPE_GRPH_STEREO) {
                REG_UPDATE(DCSURF_FLIP_CONTROL, 
SURFACE_FLIP_MODE_FOR_STEREOSYNC, 0x1);
diff --git a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubp.h 
b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubp.h
index f790ab9db6eb..d5acc348be22 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubp.h
+++ b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hubp.h
@@ -260,8 +260,7 @@ void hubp2_program_deadline(
 bool hubp2_program_surface_flip_and_addr(
        struct hubp *hubp,
        const struct dc_plane_address *address,
-       bool flip_immediate,
-       uint8_t vmid);
+       bool flip_immediate);
 
 void hubp2_program_surface_config(
        struct hubp *hubp,
diff --git a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hwseq.c 
b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hwseq.c
index d55b15fbfe99..20bc3b9fe879 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hwseq.c
+++ b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_hwseq.c
@@ -1122,15 +1122,13 @@ void dcn20_enable_plane(
                print_rq_dlg_ttu(dc, pipe_ctx);
        }
 */
-       if (dc->vm_config.valid) {
+       if (dc->vm_pa_config.valid) {
                struct vm_system_aperture_param apt;
 
                apt.sys_default.quad_part = 0;
-               apt.sys_high.quad_part = 0;
-               apt.sys_low.quad_part = 0;
 
-               apt.sys_high.quad_part = 
dc->vm_config.pa_config.system_aperture.start_addr;
-               apt.sys_low.quad_part = 
dc->vm_config.pa_config.system_aperture.end_addr;
+               apt.sys_high.quad_part = 
dc->vm_pa_config.system_aperture.start_addr;
+               apt.sys_low.quad_part = 
dc->vm_pa_config.system_aperture.end_addr;
 
                // Program system aperture settings
                
pipe_ctx->plane_res.hubp->funcs->hubp_set_vm_system_aperture_settings(pipe_ctx->plane_res.hubp,
 &apt);
@@ -1565,26 +1563,43 @@ void dcn20_disable_stream(struct pipe_ctx *pipe_ctx, 
int option)
        dce110_disable_stream(pipe_ctx, option);
 }
 
-static void dcn20_init_dchub(struct dce_hwseq *hws, struct dc *dc, struct 
dc_addr_space_config *config)
+static void dcn20_init_vm_ctx(
+               struct dce_hwseq *hws,
+               struct dc *dc,
+               struct dc_virtual_addr_space_config *va_config,
+               int vmid)
 {
-       struct hubbub_addr_config hubbub_config;
-
-       hubbub_config.pa_config.system_aperture.fb_top = 
config->pa_config.system_aperture.fb_top;
-       hubbub_config.pa_config.system_aperture.fb_offset = 
config->pa_config.system_aperture.fb_offset;
-       hubbub_config.pa_config.system_aperture.fb_base = 
config->pa_config.system_aperture.fb_base;
-       hubbub_config.pa_config.system_aperture.agp_top = 
config->pa_config.system_aperture.agp_top;
-       hubbub_config.pa_config.system_aperture.agp_bot = 
config->pa_config.system_aperture.agp_bot;
-       hubbub_config.pa_config.system_aperture.agp_base = 
config->pa_config.system_aperture.agp_base;
-       hubbub_config.pa_config.gart_config.page_table_start_addr = 
config->pa_config.gart_config.page_table_start_addr;
-       hubbub_config.pa_config.gart_config.page_table_end_addr = 
config->pa_config.gart_config.page_table_end_addr;
-       hubbub_config.pa_config.gart_config.page_table_base_addr = 
config->pa_config.gart_config.page_table_base_addr;
-
-       hubbub_config.va_config.page_table_start_addr = 
config->va_config.page_table_start_addr;
-       hubbub_config.va_config.page_table_end_addr = 
config->va_config.page_table_end_addr;
-       hubbub_config.va_config.page_table_block_size = 
config->va_config.page_table_block_size_in_bytes;
-       hubbub_config.va_config.page_table_depth = 
config->va_config.page_table_depth;
-
-       dc->res_pool->hubbub->funcs->init_dchub(dc->res_pool->hubbub, 
&hubbub_config);
+       struct dcn_hubbub_virt_addr_config config;
+
+       if (vmid == 0) {
+               ASSERT(0); /* VMID cannot be 0 for vm context */
+               return;
+       }
+
+       config.page_table_start_addr = va_config->page_table_start_addr;
+       config.page_table_end_addr = va_config->page_table_end_addr;
+       config.page_table_block_size = 
va_config->page_table_block_size_in_bytes;
+       config.page_table_depth = va_config->page_table_depth;
+       config.page_table_base_addr = va_config->page_table_base_addr;
+
+       dc->res_pool->hubbub->funcs->init_vm_ctx(dc->res_pool->hubbub, &config, 
vmid);
+}
+
+static int dcn20_init_sys_ctx(struct dce_hwseq *hws, struct dc *dc, struct 
dc_phy_addr_space_config *pa_config)
+{
+       struct dcn_hubbub_phys_addr_config config;
+
+       config.system_aperture.fb_top = pa_config->system_aperture.fb_top;
+       config.system_aperture.fb_offset = pa_config->system_aperture.fb_offset;
+       config.system_aperture.fb_base = pa_config->system_aperture.fb_base;
+       config.system_aperture.agp_top = pa_config->system_aperture.agp_top;
+       config.system_aperture.agp_bot = pa_config->system_aperture.agp_bot;
+       config.system_aperture.agp_base = pa_config->system_aperture.agp_base;
+       config.gart_config.page_table_start_addr = 
pa_config->gart_config.page_table_start_addr;
+       config.gart_config.page_table_end_addr = 
pa_config->gart_config.page_table_end_addr;
+       config.gart_config.page_table_base_addr = 
pa_config->gart_config.page_table_base_addr;
+
+       return 
dc->res_pool->hubbub->funcs->init_dchub_sys_ctx(dc->res_pool->hubbub, &config);
 }
 
 static bool patch_address_for_sbs_tb_stereo(
@@ -1619,30 +1634,19 @@ static void dcn20_update_plane_addr(const struct dc 
*dc, struct pipe_ctx *pipe_c
        bool addr_patched = false;
        PHYSICAL_ADDRESS_LOC addr;
        struct dc_plane_state *plane_state = pipe_ctx->plane_state;
-       uint8_t vmid;
 
        if (plane_state == NULL)
                return;
 
        addr_patched = patch_address_for_sbs_tb_stereo(pipe_ctx, &addr);
 
-       // Call Helper to assign correct VMID to this PTB
-       vmid = get_vmid_for_ptb(dc->vm_helper,
-                       plane_state->address.page_table_base.quad_part,
-                       pipe_ctx->pipe_idx);
-
-       // Call hubbub to program PTB of VMID only if its VA
-       // PA PTB is a one-time setup at init
-       if (vmid > 0 && dc->res_pool->hubbub->funcs->setup_vmid_ptb)
-               
dc->res_pool->hubbub->funcs->setup_vmid_ptb(dc->res_pool->hubbub,
-                               plane_state->address.page_table_base.quad_part,
-                               vmid);
+       // Call Helper to track VMID use
+       vm_helper_mark_vmid_used(dc->vm_helper, plane_state->address.vmid, 
pipe_ctx->plane_res.hubp->inst);
 
        pipe_ctx->plane_res.hubp->funcs->hubp_program_surface_flip_and_addr(
                        pipe_ctx->plane_res.hubp,
                        &plane_state->address,
-                       plane_state->flip_immediate,
-                       vmid);
+                       plane_state->flip_immediate);
 
        plane_state->status.requested_address = plane_state->address;
 
@@ -1991,7 +1995,8 @@ void dcn20_hw_sequencer_construct(struct dc *dc)
        dc->hwss.blank_pixel_data = dcn20_blank_pixel_data;
        dc->hwss.dmdata_status_done = dcn20_dmdata_status_done;
        dc->hwss.disable_stream = dcn20_disable_stream;
-       dc->hwss.init_dchub = dcn20_init_dchub;
+       dc->hwss.init_sys_ctx = dcn20_init_sys_ctx;
+       dc->hwss.init_vm_ctx = dcn20_init_vm_ctx;
        dc->hwss.disable_stream_gating = dcn20_disable_stream_gating;
        dc->hwss.enable_stream_gating = dcn20_enable_stream_gating;
        dc->hwss.setup_vupdate_interrupt = dcn20_setup_vupdate_interrupt;
diff --git a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_resource.c 
b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_resource.c
index 26af860df7d4..4cc66c2ccb1d 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_resource.c
+++ b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_resource.c
@@ -2899,7 +2899,7 @@ static bool construct(
 
        // Init the vm_helper
        if (dc->vm_helper)
-               init_vm_helper(dc->vm_helper, 16, pool->base.pipe_count);
+               vm_helper_init(dc->vm_helper, 16);
 
        /*************************************************
         *  Create resources                             *
diff --git a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_vmid.c 
b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_vmid.c
index 50953c0bd4e3..27679ef6ebe8 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_vmid.c
+++ b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_vmid.c
@@ -51,12 +51,9 @@ void dcn20_vmid_setup(struct dcn20_vmid *vmid, const struct 
dcn_vmid_page_table_
        REG_SET_2(CNTL, 0,
                        VM_CONTEXT0_PAGE_TABLE_DEPTH, config->depth,
                        VM_CONTEXT0_PAGE_TABLE_BLOCK_SIZE, config->block_size);
-}
 
-void dcn20_vmid_set_ptb(struct dcn20_vmid *vmid, uint64_t base)
-{
        REG_SET(PAGE_TABLE_BASE_ADDR_HI32, 0,
-                       VM_CONTEXT0_PAGE_DIRECTORY_ENTRY_HI32, (base >> 32) & 
0xFFFFFFFF);
+                       VM_CONTEXT0_PAGE_DIRECTORY_ENTRY_HI32, 
(config->page_table_base_addr >> 32) & 0xFFFFFFFF);
        REG_SET(PAGE_TABLE_BASE_ADDR_LO32, 0,
-                       VM_CONTEXT0_PAGE_DIRECTORY_ENTRY_LO32, base & 
0xFFFFFFFF);
+                       VM_CONTEXT0_PAGE_DIRECTORY_ENTRY_LO32, 
config->page_table_base_addr & 0xFFFFFFFF);
 }
diff --git a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_vmid.h 
b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_vmid.h
index 9c2f7016af2a..02fafb013fc6 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_vmid.h
+++ b/drivers/gpu/drm/amd/display/dc/dcn20/dcn20_vmid.h
@@ -86,6 +86,5 @@ struct dcn20_vmid {
 };
 
 void dcn20_vmid_setup(struct dcn20_vmid *vmid, const struct 
dcn_vmid_page_table_config *config);
-void dcn20_vmid_set_ptb(struct dcn20_vmid *vmid, uint64_t base);
 
 #endif /* DAL_DC_DCN20_DCN20_VMID_H_ */
diff --git a/drivers/gpu/drm/amd/display/dc/inc/hw/dchubbub.h 
b/drivers/gpu/drm/amd/display/dc/inc/hw/dchubbub.h
index d56fd7d87bbc..959f5b654611 100644
--- a/drivers/gpu/drm/amd/display/dc/inc/hw/dchubbub.h
+++ b/drivers/gpu/drm/amd/display/dc/inc/hw/dchubbub.h
@@ -87,6 +87,7 @@ struct dcn_hubbub_virt_addr_config {
        uint64_t                                page_table_end_addr;
        enum dcn_hubbub_page_table_block_size   page_table_block_size;
        enum dcn_hubbub_page_table_depth        page_table_depth;
+       uint64_t                                page_table_base_addr;
 };
 
 struct hubbub_addr_config {
@@ -105,18 +106,13 @@ struct hubbub_funcs {
                        struct dchub_init_data *dh_data);
 
 #ifdef CONFIG_DRM_AMD_DC_DCN2_0
-       void (*init_dchub)(
+       int (*init_dchub_sys_ctx)(
                        struct hubbub *hubbub,
-                       struct hubbub_addr_config *config);
-       void (*setup_vmid_ptb)(
+                       struct dcn_hubbub_phys_addr_config *pa_config);
+       void (*init_vm_ctx)(
                        struct hubbub *hubbub,
-                       uint64_t ptb,
-                       uint8_t vmid);
-
-       void (*set_ptb)(
-                       struct hubbub *hubbub,
-                       uint8_t vmid,
-                       uint64_t base_addr);
+                       struct dcn_hubbub_virt_addr_config *va_config,
+                       int vmid);
 
 #endif
        bool (*get_dcc_compression_cap)(struct hubbub *hubbub,
diff --git a/drivers/gpu/drm/amd/display/dc/inc/hw/hubp.h 
b/drivers/gpu/drm/amd/display/dc/inc/hw/hubp.h
index 342477822dc0..51bff8717cc9 100644
--- a/drivers/gpu/drm/amd/display/dc/inc/hw/hubp.h
+++ b/drivers/gpu/drm/amd/display/dc/inc/hw/hubp.h
@@ -83,8 +83,7 @@ struct hubp_funcs {
        bool (*hubp_program_surface_flip_and_addr)(
                struct hubp *hubp,
                const struct dc_plane_address *address,
-               bool flip_immediate,
-               uint8_t vmid);
+               bool flip_immediate);
 
        void (*hubp_program_pte_vm)(
                struct hubp *hubp,
diff --git a/drivers/gpu/drm/amd/display/dc/inc/hw/vmid.h 
b/drivers/gpu/drm/amd/display/dc/inc/hw/vmid.h
index 037beb0a2a27..76de0e4284e0 100644
--- a/drivers/gpu/drm/amd/display/dc/inc/hw/vmid.h
+++ b/drivers/gpu/drm/amd/display/dc/inc/hw/vmid.h
@@ -44,6 +44,7 @@ struct dcn_vmid_page_table_config {
        uint64_t        page_table_end_addr;
        enum dcn_hubbub_page_table_depth        depth;
        enum dcn_hubbub_page_table_block_size   block_size;
+       uint64_t        page_table_base_addr;
 };
 
 #endif /* DAL_DC_INC_HW_VMID_H_ */
diff --git a/drivers/gpu/drm/amd/display/dc/inc/hw_sequencer.h 
b/drivers/gpu/drm/amd/display/dc/inc/hw_sequencer.h
index 13b113d0fe19..4ffe42c27c3e 100644
--- a/drivers/gpu/drm/amd/display/dc/inc/hw_sequencer.h
+++ b/drivers/gpu/drm/amd/display/dc/inc/hw_sequencer.h
@@ -75,7 +75,8 @@ struct resource_pool;
 struct resource_context;
 struct stream_resource;
 #ifdef CONFIG_DRM_AMD_DC_DCN2_0
-struct dc_addr_space_config;
+struct dc_phy_addr_space_config;
+struct dc_virtual_addr_space_config;
 #endif
 
 struct hw_sequencer_funcs {
@@ -132,11 +133,15 @@ struct hw_sequencer_funcs {
                struct dchub_init_data *dh_data);
 
 #ifdef CONFIG_DRM_AMD_DC_DCN2_0
-       void (*init_dchub)(
-               struct dce_hwseq *hws,
-               struct dc *dc,
-               struct dc_addr_space_config *dh_data);
-
+       int (*init_sys_ctx)(
+                       struct dce_hwseq *hws,
+                       struct dc *dc,
+                       struct dc_phy_addr_space_config *pa_config);
+       void (*init_vm_ctx)(
+                       struct dce_hwseq *hws,
+                       struct dc *dc,
+                       struct dc_virtual_addr_space_config *va_config,
+                       int vmid);
 #endif
        void (*update_mpcc)(
                struct dc *dc,
diff --git a/drivers/gpu/drm/amd/display/dc/inc/vm_helper.h 
b/drivers/gpu/drm/amd/display/dc/inc/vm_helper.h
index 193407f76a80..8bfcef0a3675 100644
--- a/drivers/gpu/drm/amd/display/dc/inc/vm_helper.h
+++ b/drivers/gpu/drm/amd/display/dc/inc/vm_helper.h
@@ -28,29 +28,21 @@
 
 #include "dc_types.h"
 
-#define MAX_VMID 16
 #define MAX_HUBP 6
 
 struct vmid_usage {
-       uint16_t vmid_usage[2];
+       int vmid_usage[2];
 };
 
 struct vm_helper {
        unsigned int num_vmid;
-       unsigned int num_hubp;
-       unsigned int num_vmids_available;
-       uint64_t ptb_assigned_to_vmid[MAX_VMID];
        struct vmid_usage hubp_vmid_usage[MAX_HUBP];
 };
 
-uint8_t get_vmid_for_ptb(
-               struct vm_helper *vm_helper,
-               int64_t ptb,
-               uint8_t pipe_idx);
+void vm_helper_mark_vmid_used(struct vm_helper *vm_helper, unsigned int pos, 
uint8_t hubp_idx);
 
-void init_vm_helper(
+void vm_helper_init(
        struct vm_helper *vm_helper,
-       unsigned int num_vmid,
-       unsigned int num_hubp);
+       unsigned int num_vmid);
 
 #endif /* DC_INC_VM_HELPER_H_ */
diff --git a/drivers/gpu/drm/amd/display/modules/inc/mod_vmid.h 
b/drivers/gpu/drm/amd/display/modules/inc/mod_vmid.h
new file mode 100644
index 000000000000..c7672f3b837f
--- /dev/null
+++ b/drivers/gpu/drm/amd/display/modules/inc/mod_vmid.h
@@ -0,0 +1,44 @@
+/*
+ * Copyright 2019 Advanced Micro Devices, Inc.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
+ * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+ * OTHER DEALINGS IN THE SOFTWARE.
+ *
+ * Authors: AMD
+ *
+ */
+
+#ifndef MOD_VMID_H_
+#define MOD_VMID_H_
+
+#define MAX_VMID 16
+
+#include "dc.h"
+
+struct mod_vmid {
+       int dummy;
+};
+
+uint8_t mod_vmid_get_for_ptb(struct mod_vmid *mod_vmid, uint64_t ptb);
+void mod_vmid_reset(struct mod_vmid *mod_vmid);
+struct mod_vmid *mod_vmid_create(
+               struct dc *dc,
+               unsigned int num_vmid,
+               struct dc_virtual_addr_space_config *va_config);
+
+#endif /* MOD_VMID_H_ */
diff --git a/drivers/gpu/drm/amd/display/modules/vmid/vmid.c 
b/drivers/gpu/drm/amd/display/modules/vmid/vmid.c
new file mode 100644
index 000000000000..431052626d93
--- /dev/null
+++ b/drivers/gpu/drm/amd/display/modules/vmid/vmid.c
@@ -0,0 +1,158 @@
+/*
+ * Copyright 2019 Advanced Micro Devices, Inc.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
+ * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+ * OTHER DEALINGS IN THE SOFTWARE.
+ *
+ * Authors: AMD
+ *
+ */
+
+#include "mod_vmid.h"
+
+struct core_vmid {
+       struct mod_vmid public;
+       struct dc *dc;
+
+       unsigned int num_vmid;
+       unsigned int num_vmids_available;
+       uint64_t ptb_assigned_to_vmid[MAX_VMID];
+       struct dc_virtual_addr_space_config base_config;
+};
+
+#define MOD_VMID_TO_CORE(mod_vmid)\
+               container_of(mod_vmid, struct core_vmid, public)
+
+static void add_ptb_to_table(struct core_vmid *core_vmid, unsigned int vmid, 
uint64_t ptb)
+{
+       core_vmid->ptb_assigned_to_vmid[vmid] = ptb;
+       core_vmid->num_vmids_available--;
+}
+
+static void clear_entry_from_vmid_table(struct core_vmid *core_vmid, unsigned 
int vmid)
+{
+       core_vmid->ptb_assigned_to_vmid[vmid] = 0;
+       core_vmid->num_vmids_available++;
+}
+
+static void evict_vmids(struct core_vmid *core_vmid)
+{
+       int i;
+       uint16_t ord = dc_get_vmid_use_vector(core_vmid->dc);
+
+       // At this point any positions with value 0 are unused vmids, evict them
+       for (i = 1; i < core_vmid->num_vmid; i++) {
+               if (ord & (1u << i))
+                       clear_entry_from_vmid_table(core_vmid, i);
+       }
+}
+
+// Return value of -1 indicates vmid table unitialized or ptb dne in the table
+static int get_existing_vmid_for_ptb(struct core_vmid *core_vmid, uint64_t ptb)
+{
+       int i;
+
+       for (i = 0; i < core_vmid->num_vmid; i++) {
+               if (core_vmid->ptb_assigned_to_vmid[i] == ptb)
+                       return i;
+       }
+
+       return -1;
+}
+
+// Expected to be called only when there's an available vmid
+static int get_next_available_vmid(struct core_vmid *core_vmid)
+{
+       int i;
+
+       for (i = 1; i < core_vmid->num_vmid; i++) {
+               if (core_vmid->ptb_assigned_to_vmid[i] == 0)
+                       return i;
+       }
+
+       return -1;
+}
+
+uint8_t mod_vmid_get_for_ptb(struct mod_vmid *mod_vmid, uint64_t ptb)
+{
+       struct core_vmid *core_vmid = MOD_VMID_TO_CORE(mod_vmid);
+       unsigned int vmid = 0;
+
+       // Physical address gets vmid 0
+       if (ptb == 0)
+               return 0;
+
+       vmid = get_existing_vmid_for_ptb(core_vmid, ptb);
+
+       if (vmid == -1) {
+               struct dc_virtual_addr_space_config va_config = 
core_vmid->base_config;
+
+               va_config.page_table_base_addr = ptb;
+
+               if (core_vmid->num_vmids_available == 0)
+                       evict_vmids(core_vmid);
+
+               vmid = get_next_available_vmid(core_vmid);
+               add_ptb_to_table(core_vmid, vmid, ptb);
+
+               dc_setup_vm_context(core_vmid->dc, &va_config, vmid);
+       }
+
+       return vmid;
+}
+
+void mod_vmid_reset(struct mod_vmid *mod_vmid)
+{
+       struct core_vmid *core_vmid = MOD_VMID_TO_CORE(mod_vmid);
+
+       core_vmid->num_vmids_available = core_vmid->num_vmid - 1;
+       memset(core_vmid->ptb_assigned_to_vmid, 0, 
sizeof(core_vmid->ptb_assigned_to_vmid[0]) * MAX_VMID);
+}
+
+struct mod_vmid *mod_vmid_create(
+               struct dc *dc,
+               unsigned int num_vmid,
+               struct dc_virtual_addr_space_config *va_config)
+{
+       struct core_vmid *core_vmid;
+
+       if (num_vmid <= 1)
+               goto fail_no_vm_ctx;
+
+       if (dc == NULL)
+               goto fail_dc_null;
+
+       core_vmid = kzalloc(sizeof(struct core_vmid), GFP_KERNEL);
+
+       if (core_vmid == NULL)
+               goto fail_alloc_context;
+
+       core_vmid->dc = dc;
+       core_vmid->num_vmid = num_vmid;
+       core_vmid->num_vmids_available = num_vmid - 1;
+       core_vmid->base_config = *va_config;
+
+       memset(core_vmid->ptb_assigned_to_vmid, 0, 
sizeof(core_vmid->ptb_assigned_to_vmid[0]) * MAX_VMID);
+
+       return &core_vmid->public;
+
+fail_no_vm_ctx:
+fail_alloc_context:
+fail_dc_null:
+       return NULL;
+}
-- 
2.20.1

_______________________________________________
amd-gfx mailing list
amd-gfx@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/amd-gfx

Reply via email to