[Mesa-dev] [PATCH 20/20] anv/pipeline: Use get_scratch_space/address for compute shaders

[Jason Ekstrand]

---
 src/intel/vulkan/genX_pipeline.c | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/src/intel/vulkan/genX_pipeline.c b/src/intel/vulkan/genX_pipeline.c
index 0af37e4..a1beb81 100644
--- a/src/intel/vulkan/genX_pipeline.c
+++ b/src/intel/vulkan/genX_pipeline.c
@@ -1452,14 +1452,10 @@ compute_pipeline_create(
 
    const uint32_t subslices = MAX2(physical_device->subslice_total, 1);
 
+   const struct anv_shader_bin *cs_bin =
+      pipeline->shaders[MESA_SHADER_COMPUTE];
+
    anv_batch_emit(&pipeline->batch, GENX(MEDIA_VFE_STATE), vfe) {
-      vfe.ScratchSpaceBasePointer = (struct anv_address) {
-         .bo = anv_scratch_pool_alloc(device, &device->scratch_pool,
-                                      MESA_SHADER_COMPUTE,
-                                      cs_prog_data->base.total_scratch),
-         .offset = 0,
-      };
-      vfe.PerThreadScratchSpace  = ffs(cs_prog_data->base.total_scratch / 2048);
 #if GEN_GEN > 7
       vfe.StackSize              = 0;
 #else
@@ -1474,10 +1470,12 @@ compute_pipeline_create(
 #endif
       vfe.URBEntryAllocationSize = GEN_GEN <= 7 ? 0 : 2;
       vfe.CURBEAllocationSize    = vfe_curbe_allocation;
+
+      vfe.PerThreadScratchSpace = get_scratch_space(cs_bin);
+      vfe.ScratchSpaceBasePointer =
+         get_scratch_address(pipeline, MESA_SHADER_COMPUTE, cs_bin);
    }
 
-   const struct anv_shader_bin *cs_bin =
-      pipeline->shaders[MESA_SHADER_COMPUTE];
    struct GENX(INTERFACE_DESCRIPTOR_DATA) desc = {
       .KernelStartPointer     = cs_bin->kernel.offset,
 
-- 
2.5.0.400.gff86faf