xref: /aosp_15_r20/external/mesa3d/src/gallium/auxiliary/draw/draw_pt_fetch_shade_pipeline_llvm.c (revision 6104692788411f58d303aa86923a9ff6ecaded22)
1 /**************************************************************************
2  *
3  * Copyright 2010 VMware, Inc.
4  * All Rights Reserved.
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a
7  * copy of this software and associated documentation files (the
8  * "Software"), to deal in the Software without restriction, including
9  * without limitation the rights to use, copy, modify, merge, publish,
10  * distribute, sub license, and/or sell copies of the Software, and to
11  * permit persons to whom the Software is furnished to do so, subject to
12  * the following conditions:
13  *
14  * The above copyright notice and this permission notice (including the
15  * next paragraph) shall be included in all copies or substantial portions
16  * of the Software.
17  *
18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25  *
26  **************************************************************************/
27 
28 #include "util/u_math.h"
29 #include "util/u_memory.h"
30 #include "util/u_prim.h"
31 #include "draw/draw_context.h"
32 #include "draw/draw_gs.h"
33 #include "draw/draw_tess.h"
34 #include "draw/draw_vbuf.h"
35 #include "draw/draw_vertex.h"
36 #include "draw/draw_pt.h"
37 #include "draw/draw_prim_assembler.h"
38 #include "draw/draw_vs.h"
39 #include "draw/draw_llvm.h"
40 #include "gallivm/lp_bld_init.h"
41 #include "gallivm/lp_bld_debug.h"
42 
43 
44 struct llvm_middle_end {
45    struct draw_pt_middle_end base;
46    struct draw_context *draw;
47 
48    struct pt_emit *emit;
49    struct pt_so_emit *so_emit;
50    struct pt_fetch *fetch;
51    struct pt_post_vs *post_vs;
52 
53 
54    unsigned vertex_data_offset;
55    unsigned vertex_size;
56    enum mesa_prim input_prim;
57    unsigned opt;
58 
59    struct draw_llvm *llvm;
60    struct draw_llvm_variant *current_variant;
61 };
62 
63 
64 /** cast wrapper */
65 static inline struct llvm_middle_end *
llvm_middle_end(struct draw_pt_middle_end * middle)66 llvm_middle_end(struct draw_pt_middle_end *middle)
67 {
68    return (struct llvm_middle_end *) middle;
69 }
70 
71 
72 static void
llvm_middle_end_prepare_gs(struct llvm_middle_end * fpme)73 llvm_middle_end_prepare_gs(struct llvm_middle_end *fpme)
74 {
75    struct draw_context *draw = fpme->draw;
76    struct draw_llvm *llvm = fpme->llvm;
77    struct draw_geometry_shader *gs = draw->gs.geometry_shader;
78    struct draw_gs_llvm_variant_list_item *li;
79    struct llvm_geometry_shader *shader = llvm_geometry_shader(gs);
80    char store[DRAW_GS_LLVM_MAX_VARIANT_KEY_SIZE];
81    struct draw_gs_llvm_variant_key *key = draw_gs_llvm_make_variant_key(llvm, store);
82 
83    /* Search shader's list of variants for the key */
84    struct draw_gs_llvm_variant *variant = NULL;
85    LIST_FOR_EACH_ENTRY(li, &shader->variants.list, list) {
86       if (memcmp(&li->base->key, key, shader->variant_key_size) == 0) {
87          variant = li->base;
88          break;
89       }
90    }
91 
92    if (variant) {
93       /* found the variant, move to head of global list (for LRU) */
94       list_move_to(&variant->list_item_global.list, &llvm->gs_variants_list.list);
95    } else {
96       /* Need to create new variant */
97 
98       /* First check if we've created too many variants.  If so, free
99        * 3.125% of the LRU to avoid using too much memory.
100        */
101       if (llvm->nr_gs_variants >= DRAW_MAX_SHADER_VARIANTS) {
102          if (gallivm_debug & GALLIVM_DEBUG_PERF) {
103             debug_printf("Evicting GS: %u gs variants,\t%u total variants\n",
104                       shader->variants_cached, llvm->nr_gs_variants);
105          }
106 
107          /*
108           * XXX: should we flush here ?
109           */
110          struct draw_gs_llvm_variant_list_item *item;
111          for (unsigned i = 0; i < DRAW_MAX_SHADER_VARIANTS / 32; i++) {
112             if (list_is_empty(&llvm->gs_variants_list.list)) {
113                break;
114             }
115             item = list_last_entry(&llvm->gs_variants_list.list,
116                                    struct draw_gs_llvm_variant_list_item, list);
117             assert(item);
118             assert(item->base);
119             draw_gs_llvm_destroy_variant(item->base);
120          }
121       }
122 
123       variant = draw_gs_llvm_create_variant(llvm, draw_total_gs_outputs(draw), key);
124 
125       if (variant) {
126          list_add(&variant->list_item_local.list, &shader->variants.list);
127          list_add(&variant->list_item_global.list, &llvm->gs_variants_list.list);
128          llvm->nr_gs_variants++;
129          shader->variants_cached++;
130       }
131    }
132 
133    gs->current_variant = variant;
134 }
135 
136 
137 static void
llvm_middle_end_prepare_tcs(struct llvm_middle_end * fpme)138 llvm_middle_end_prepare_tcs(struct llvm_middle_end *fpme)
139 {
140    struct draw_context *draw = fpme->draw;
141    struct draw_llvm *llvm = fpme->llvm;
142    struct draw_tess_ctrl_shader *tcs = draw->tcs.tess_ctrl_shader;
143    struct draw_tcs_llvm_variant_list_item *li;
144    struct llvm_tess_ctrl_shader *shader = llvm_tess_ctrl_shader(tcs);
145    char store[DRAW_TCS_LLVM_MAX_VARIANT_KEY_SIZE];
146    const struct draw_tcs_llvm_variant_key *key =
147       draw_tcs_llvm_make_variant_key(llvm, store);
148 
149    /* Search shader's list of variants for the key */
150    struct draw_tcs_llvm_variant *variant = NULL;
151    LIST_FOR_EACH_ENTRY(li, &shader->variants.list, list) {
152       if (memcmp(&li->base->key, key, shader->variant_key_size) == 0) {
153          variant = li->base;
154          break;
155       }
156    }
157 
158    if (variant) {
159       /* found the variant, move to head of global list (for LRU) */
160       list_move_to(&variant->list_item_global.list,
161                    &llvm->tcs_variants_list.list);
162    } else {
163       /* Need to create new variant */
164 
165       /* First check if we've created too many variants.  If so, free
166        * 3.125% of the LRU to avoid using too much memory.
167        */
168       if (llvm->nr_tcs_variants >= DRAW_MAX_SHADER_VARIANTS) {
169          if (gallivm_debug & GALLIVM_DEBUG_PERF) {
170             debug_printf("Evicting TCS: %u tcs variants,\t%u total variants\n",
171                       shader->variants_cached, llvm->nr_tcs_variants);
172          }
173 
174          /*
175           * XXX: should we flush here ?
176           */
177          for (unsigned i = 0; i < DRAW_MAX_SHADER_VARIANTS / 32; i++) {
178             struct draw_tcs_llvm_variant_list_item *item;
179             if (list_is_empty(&llvm->tcs_variants_list.list)) {
180                break;
181             }
182             item = list_last_entry(&llvm->tcs_variants_list.list,
183                                    struct draw_tcs_llvm_variant_list_item, list);
184             assert(item);
185             assert(item->base);
186             draw_tcs_llvm_destroy_variant(item->base);
187          }
188       }
189 
190       variant = draw_tcs_llvm_create_variant(llvm, 0, key);
191 
192       if (variant) {
193          list_add(&variant->list_item_local.list, &shader->variants.list);
194          list_add(&variant->list_item_global.list, &llvm->tcs_variants_list.list);
195          llvm->nr_tcs_variants++;
196          shader->variants_cached++;
197       }
198    }
199 
200    tcs->current_variant = variant;
201 }
202 
203 
204 static void
llvm_middle_end_prepare_tes(struct llvm_middle_end * fpme)205 llvm_middle_end_prepare_tes(struct llvm_middle_end *fpme)
206 {
207    struct draw_context *draw = fpme->draw;
208    struct draw_llvm *llvm = fpme->llvm;
209    struct draw_tess_eval_shader *tes = draw->tes.tess_eval_shader;
210    struct draw_tes_llvm_variant *variant = NULL;
211    struct draw_tes_llvm_variant_list_item *li;
212    struct llvm_tess_eval_shader *shader = llvm_tess_eval_shader(tes);
213    char store[DRAW_TES_LLVM_MAX_VARIANT_KEY_SIZE];
214    const struct draw_tes_llvm_variant_key *key =
215       draw_tes_llvm_make_variant_key(llvm, store);
216 
217    /* Search shader's list of variants for the key */
218    LIST_FOR_EACH_ENTRY(li, &shader->variants.list, list) {
219       if (memcmp(&li->base->key, key, shader->variant_key_size) == 0) {
220          variant = li->base;
221          break;
222       }
223    }
224 
225    if (variant) {
226       /* found the variant, move to head of global list (for LRU) */
227       list_move_to(&variant->list_item_global.list,
228                    &llvm->tes_variants_list.list);
229    } else {
230       /* Need to create new variant */
231 
232       /* First check if we've created too many variants.  If so, free
233        * 3.125% of the LRU to avoid using too much memory.
234        */
235       if (llvm->nr_tes_variants >= DRAW_MAX_SHADER_VARIANTS) {
236          if (gallivm_debug & GALLIVM_DEBUG_PERF) {
237             debug_printf("Evicting TES: %u tes variants,\t%u total variants\n",
238                       shader->variants_cached, llvm->nr_tes_variants);
239          }
240 
241          /*
242           * XXX: should we flush here ?
243           */
244          for (unsigned i = 0; i < DRAW_MAX_SHADER_VARIANTS / 32; i++) {
245             struct draw_tes_llvm_variant_list_item *item;
246             if (list_is_empty(&llvm->tes_variants_list.list)) {
247                break;
248             }
249             item = list_last_entry(&llvm->tes_variants_list.list,
250                                    struct draw_tes_llvm_variant_list_item, list);
251             assert(item);
252             assert(item->base);
253             draw_tes_llvm_destroy_variant(item->base);
254          }
255       }
256 
257       variant = draw_tes_llvm_create_variant(llvm, draw_total_tes_outputs(draw), key);
258 
259       if (variant) {
260          list_add(&variant->list_item_local.list, &shader->variants.list);
261          list_add(&variant->list_item_global.list, &llvm->tes_variants_list.list);
262          llvm->nr_tes_variants++;
263          shader->variants_cached++;
264       }
265    }
266 
267    tes->current_variant = variant;
268 }
269 
270 
271 /**
272  * Prepare/validate middle part of the vertex pipeline.
273  * NOTE: if you change this function, also look at the non-LLVM
274  * function fetch_pipeline_prepare() for similar changes.
275  */
276 static void
llvm_middle_end_prepare(struct draw_pt_middle_end * middle,enum mesa_prim in_prim,unsigned opt,unsigned * max_vertices)277 llvm_middle_end_prepare(struct draw_pt_middle_end *middle,
278                         enum mesa_prim in_prim,
279                         unsigned opt,
280                         unsigned *max_vertices)
281 {
282    struct llvm_middle_end *fpme = llvm_middle_end(middle);
283    struct draw_context *draw = fpme->draw;
284    struct draw_llvm *llvm = fpme->llvm;
285    struct draw_vertex_shader *vs = draw->vs.vertex_shader;
286    struct draw_geometry_shader *gs = draw->gs.geometry_shader;
287    struct draw_tess_ctrl_shader *tcs = draw->tcs.tess_ctrl_shader;
288    struct draw_tess_eval_shader *tes = draw->tes.tess_eval_shader;
289    const enum mesa_prim out_prim =
290       gs ? gs->output_primitive : tes ? get_tes_output_prim(tes) :
291       u_assembled_prim(in_prim);
292    unsigned point_line_clip = draw->rasterizer->fill_front == PIPE_POLYGON_MODE_POINT ||
293                               draw->rasterizer->fill_front == PIPE_POLYGON_MODE_LINE ||
294                               out_prim == MESA_PRIM_POINTS ||
295                               u_reduced_prim(out_prim) == MESA_PRIM_LINES;
296 
297    fpme->input_prim = in_prim;
298    fpme->opt = opt;
299 
300    draw_pt_post_vs_prepare(fpme->post_vs,
301                            draw->clip_xy,
302                            draw->clip_z,
303                            draw->clip_user,
304                            point_line_clip ? draw->guard_band_points_lines_xy :
305                                              draw->guard_band_xy,
306                            draw->bypass_viewport,
307                            draw->rasterizer->clip_halfz,
308                            (draw->vs.edgeflag_output ? true : false));
309 
310    draw_pt_so_emit_prepare(fpme->so_emit, (gs == NULL && tes == NULL));
311 
312    if (!(opt & PT_PIPELINE)) {
313       draw_pt_emit_prepare(fpme->emit, out_prim, max_vertices);
314 
315       *max_vertices = MAX2(*max_vertices, 4096);
316    } else {
317       /* limit max fetches by limiting max_vertices */
318       *max_vertices = 4096;
319    }
320 
321    /* Get the number of float[4] attributes per vertex.
322     * Note: this must be done after draw_pt_emit_prepare() since that
323     * can effect the vertex size.
324     */
325    unsigned nr = MAX2(vs->info.num_inputs, draw_total_vs_outputs(draw));
326 
327    /* Always leave room for the vertex header whether we need it or
328     * not.  It's hard to get rid of it in particular because of the
329     * viewport code in draw_pt_post_vs.c.
330     */
331    fpme->vertex_size = sizeof(struct vertex_header) + nr * 4 * sizeof(float);
332 
333    /* return even number */
334    *max_vertices = *max_vertices & ~1;
335 
336    /* Find/create the vertex shader variant */
337    {
338       struct draw_llvm_variant *variant = NULL;
339       struct draw_llvm_variant_list_item *li;
340       struct llvm_vertex_shader *shader = llvm_vertex_shader(vs);
341       char store[DRAW_LLVM_MAX_VARIANT_KEY_SIZE];
342       struct draw_llvm_variant_key *key = draw_llvm_make_variant_key(llvm, store);
343 
344       /* Search shader's list of variants for the key */
345       LIST_FOR_EACH_ENTRY(li, &shader->variants.list, list) {
346          if (memcmp(&li->base->key, key, shader->variant_key_size) == 0) {
347             variant = li->base;
348             break;
349          }
350       }
351 
352       if (variant) {
353          /* found the variant, move to head of global list (for LRU) */
354          list_move_to(&variant->list_item_global.list, &llvm->vs_variants_list.list);
355       } else {
356          /* Need to create new variant */
357 
358          /* First check if we've created too many variants.  If so, free
359           * 3.125% of the LRU to avoid using too much memory.
360           */
361          if (llvm->nr_variants >= DRAW_MAX_SHADER_VARIANTS) {
362             if (gallivm_debug & GALLIVM_DEBUG_PERF) {
363                debug_printf("Evicting VS: %u vs variants,\t%u total variants\n",
364                          shader->variants_cached, llvm->nr_variants);
365             }
366 
367             /*
368              * XXX: should we flush here ?
369              */
370             for (unsigned i = 0; i < DRAW_MAX_SHADER_VARIANTS / 32; i++) {
371                struct draw_llvm_variant_list_item *item;
372                if (list_is_empty(&llvm->vs_variants_list.list)) {
373                   break;
374                }
375                item = list_last_entry(&llvm->vs_variants_list.list,
376                                     struct draw_llvm_variant_list_item, list);
377                assert(item);
378                assert(item->base);
379                draw_llvm_destroy_variant(item->base);
380             }
381          }
382 
383          variant = draw_llvm_create_variant(llvm, nr, key);
384 
385          if (variant) {
386             list_add(&variant->list_item_local.list, &shader->variants.list);
387             list_add(&variant->list_item_global.list, &llvm->vs_variants_list.list);
388             llvm->nr_variants++;
389             shader->variants_cached++;
390          }
391       }
392 
393       fpme->current_variant = variant;
394    }
395 
396    if (gs) {
397       llvm_middle_end_prepare_gs(fpme);
398    }
399    if (tcs) {
400       llvm_middle_end_prepare_tcs(fpme);
401    }
402    if (tes) {
403       llvm_middle_end_prepare_tes(fpme);
404    }
405 }
406 
407 
408 static unsigned
get_num_consts_robust(struct draw_context * draw,struct draw_buffer_info * bufs,unsigned idx)409 get_num_consts_robust(struct draw_context *draw, struct draw_buffer_info *bufs, unsigned idx)
410 {
411    uint64_t const_bytes = bufs[idx].size;
412 
413    if (const_bytes < sizeof(float))
414       return 0;
415 
416    return DIV_ROUND_UP(const_bytes, draw->constant_buffer_stride);
417 }
418 
419 /**
420  * Bind/update constant buffer pointers, clip planes and viewport dims.
421  * These are "light weight" parameters which aren't baked into the
422  * generated code.  Updating these items is much cheaper than revalidating
423  * and rebuilding the generated pipeline code.
424  */
425 static void
llvm_middle_end_bind_parameters(struct draw_pt_middle_end * middle)426 llvm_middle_end_bind_parameters(struct draw_pt_middle_end *middle)
427 {
428    static const float fake_const_buf[4];
429    struct llvm_middle_end *fpme = llvm_middle_end(middle);
430    struct draw_context *draw = fpme->draw;
431    struct draw_llvm *llvm = fpme->llvm;
432    unsigned i;
433 
434    for (enum pipe_shader_type shader_type = PIPE_SHADER_VERTEX; shader_type <= PIPE_SHADER_GEOMETRY; shader_type++) {
435       for (i = 0; i < ARRAY_SIZE(llvm->jit_resources[shader_type].constants); ++i) {
436          /*
437           * There could be a potential issue with rounding this up, as the
438           * shader expects 16-byte allocations, the fix is likely to move
439           * to LOAD intrinsic in the future and remove the vec4 constraint.
440           */
441          int num_consts = get_num_consts_robust(draw, draw->pt.user.constants[shader_type], i);
442          llvm->jit_resources[shader_type].constants[i].f = draw->pt.user.constants[shader_type][i].ptr;
443          llvm->jit_resources[shader_type].constants[i].num_elements = num_consts;
444          if (num_consts == 0) {
445             llvm->jit_resources[shader_type].constants[i].f = fake_const_buf;
446          }
447       }
448       for (i = 0; i < ARRAY_SIZE(llvm->jit_resources[shader_type].ssbos); ++i) {
449          int num_ssbos = draw->pt.user.ssbos[shader_type][i].size;
450          llvm->jit_resources[shader_type].ssbos[i].u = draw->pt.user.ssbos[shader_type][i].ptr;
451          llvm->jit_resources[shader_type].ssbos[i].num_elements = num_ssbos;
452          if (num_ssbos == 0) {
453             llvm->jit_resources[shader_type].ssbos[i].u = (const uint32_t *)fake_const_buf;
454          }
455       }
456 
457       llvm->jit_resources[shader_type].aniso_filter_table = lp_build_sample_aniso_filter_table();
458    }
459 
460    llvm->vs_jit_context.planes =
461       (float (*)[DRAW_TOTAL_CLIP_PLANES][4]) draw->pt.user.planes[0];
462    llvm->gs_jit_context.planes =
463       (float (*)[DRAW_TOTAL_CLIP_PLANES][4]) draw->pt.user.planes[0];
464 
465    llvm->vs_jit_context.viewports = draw->viewports;
466    llvm->gs_jit_context.viewports = draw->viewports;
467 }
468 
469 
470 static void
pipeline(struct llvm_middle_end * llvm,const struct draw_vertex_info * vert_info,const struct draw_prim_info * prim_info)471 pipeline(struct llvm_middle_end *llvm,
472          const struct draw_vertex_info *vert_info,
473          const struct draw_prim_info *prim_info)
474 {
475    if (prim_info->linear)
476       draw_pipeline_run_linear(llvm->draw, vert_info, prim_info);
477    else
478       draw_pipeline_run(llvm->draw, vert_info, prim_info);
479 }
480 
481 
482 static void
emit(struct pt_emit * emit,const struct draw_vertex_info * vert_info,const struct draw_prim_info * prim_info)483 emit(struct pt_emit *emit,
484      const struct draw_vertex_info *vert_info,
485      const struct draw_prim_info *prim_info)
486 {
487    if (prim_info->linear)
488       draw_pt_emit_linear(emit, vert_info, prim_info);
489    else
490       draw_pt_emit(emit, vert_info, prim_info);
491 }
492 
493 
494 static void
llvm_pipeline_generic(struct draw_pt_middle_end * middle,const struct draw_fetch_info * fetch_info,const struct draw_prim_info * in_prim_info)495 llvm_pipeline_generic(struct draw_pt_middle_end *middle,
496                       const struct draw_fetch_info *fetch_info,
497                       const struct draw_prim_info *in_prim_info)
498 {
499    struct llvm_middle_end *fpme = llvm_middle_end(middle);
500    struct draw_context *draw = fpme->draw;
501    struct draw_geometry_shader *gshader = draw->gs.geometry_shader;
502    struct draw_tess_ctrl_shader *tcs_shader = draw->tcs.tess_ctrl_shader;
503    struct draw_tess_eval_shader *tes_shader = draw->tes.tess_eval_shader;
504    struct draw_prim_info tcs_prim_info;
505    struct draw_prim_info tes_prim_info;
506    struct draw_prim_info gs_prim_info[TGSI_MAX_VERTEX_STREAMS];
507    struct draw_vertex_info llvm_vert_info;
508    struct draw_vertex_info tcs_vert_info;
509    struct draw_vertex_info tes_vert_info;
510    struct draw_vertex_info *vert_info;
511    struct draw_prim_info ia_prim_info;
512    struct draw_vertex_info ia_vert_info;
513    const struct draw_prim_info *prim_info = in_prim_info;
514    bool free_prim_info = false;
515    unsigned opt = fpme->opt;
516    bool clipped = 0;
517    uint16_t *tes_elts_out = NULL;
518 
519    assert(fetch_info->count > 0);
520 
521    llvm_vert_info.count = fetch_info->count;
522    llvm_vert_info.vertex_size = fpme->vertex_size;
523    llvm_vert_info.stride = fpme->vertex_size;
524    llvm_vert_info.verts = (struct vertex_header *)
525       MALLOC(fpme->vertex_size *
526              align(fetch_info->count, lp_native_vector_width / 32) +
527              DRAW_EXTRA_VERTICES_PADDING);
528    if (!llvm_vert_info.verts) {
529       assert(0);
530       return;
531    }
532 
533    if (draw->collect_statistics) {
534       draw->statistics.ia_vertices += prim_info->count;
535       if (prim_info->prim == MESA_PRIM_PATCHES)
536          draw->statistics.ia_primitives +=
537             prim_info->count / draw->pt.vertices_per_patch;
538       else
539          draw->statistics.ia_primitives +=
540             u_decomposed_prims_for_vertices(prim_info->prim, prim_info->count);
541       draw->statistics.vs_invocations += fetch_info->count;
542    }
543 
544    {
545       unsigned start, vertex_id_offset;
546       const unsigned *elts;
547 
548       if (fetch_info->linear) {
549          start = fetch_info->start;
550          vertex_id_offset = draw->start_index;
551          elts = NULL;
552       } else {
553          start = draw->pt.user.eltMax;
554          vertex_id_offset = draw->pt.user.eltBias;
555          elts = fetch_info->elts;
556       }
557       /* Run vertex fetch shader */
558       clipped = fpme->current_variant->jit_func(&fpme->llvm->vs_jit_context,
559                                                 &fpme->llvm->jit_resources[PIPE_SHADER_VERTEX],
560                                                 llvm_vert_info.verts,
561                                                 draw->pt.user.vbuffer,
562                                                 fetch_info->count,
563                                                 start,
564                                                 fpme->vertex_size,
565                                                 draw->pt.vertex_buffer,
566                                                 draw->instance_id,
567                                                 vertex_id_offset,
568                                                 draw->start_instance,
569                                                 elts,
570                                                 draw->pt.user.drawid,
571                                                 draw->pt.user.viewid);
572 
573       /* Finished with fetch and vs */
574       fetch_info = NULL;
575       vert_info = &llvm_vert_info;
576    }
577 
578    if (opt & PT_SHADE) {
579       struct draw_vertex_shader *vshader = draw->vs.vertex_shader;
580       if (tcs_shader) {
581          draw_tess_ctrl_shader_run(tcs_shader,
582                                    vert_info,
583                                    prim_info,
584                                    &vshader->info,
585                                    &tcs_vert_info,
586                                    &tcs_prim_info);
587          FREE(vert_info->verts);
588          vert_info = &tcs_vert_info;
589          prim_info = &tcs_prim_info;
590 
591       } else if (tes_shader) {
592          unsigned num_prims = prim_info->count / draw->pt.vertices_per_patch;
593          tcs_prim_info = *prim_info;
594          tcs_prim_info.primitive_count = num_prims;
595          prim_info = &tcs_prim_info;
596       }
597 
598       if (tes_shader) {
599          draw_tess_eval_shader_run(tes_shader,
600                                    tcs_shader ? tcs_shader->vertices_out : draw->pt.vertices_per_patch,
601                                    vert_info,
602                                    prim_info,
603                                    tcs_shader ? &tcs_shader->info : &vshader->info,
604                                    &tes_vert_info,
605                                    &tes_prim_info, &tes_elts_out);
606 
607          FREE(vert_info->verts);
608          vert_info = &tes_vert_info;
609          prim_info = &tes_prim_info;
610          free_prim_info = true;
611 
612          /*
613           * pt emit can only handle ushort number of vertices (see
614           * render->allocate_vertices).
615           * vsplit guarantees there's never more than 4096, however GS can
616           * easily blow this up (by a factor of 256 (or even 1024) max).
617           */
618          if (vert_info->count > 65535) {
619             opt |= PT_PIPELINE;
620          }
621       }
622    }
623 
624    struct draw_vertex_info gs_vert_info[TGSI_MAX_VERTEX_STREAMS];
625    memset(&gs_vert_info, 0, sizeof(gs_vert_info));
626 
627    if ((opt & PT_SHADE) && gshader) {
628       struct draw_vertex_shader *vshader = draw->vs.vertex_shader;
629       draw_geometry_shader_run(gshader,
630                                draw->pt.user.constants[PIPE_SHADER_GEOMETRY],
631                                vert_info,
632                                prim_info,
633                                tes_shader ? &tes_shader->info : &vshader->info,
634                                gs_vert_info,
635                                gs_prim_info);
636 
637       FREE(vert_info->verts);
638       if (free_prim_info) {
639          FREE(prim_info->primitive_lengths);
640          FREE(tes_elts_out);
641       }
642       vert_info = &gs_vert_info[0];
643       prim_info = &gs_prim_info[0];
644       free_prim_info = false;
645       /*
646        * pt emit can only handle ushort number of vertices (see
647        * render->allocate_vertices).
648        * vsplit guarantees there's never more than 4096, however GS can
649        * easily blow this up (by a factor of 256 (or even 1024) max).
650        */
651       if (vert_info->count > 65535) {
652          opt |= PT_PIPELINE;
653       }
654    } else {
655       if (!tes_shader &&
656           draw_prim_assembler_is_required(draw, prim_info, vert_info)) {
657          draw_prim_assembler_run(draw, prim_info, vert_info,
658                                  &ia_prim_info, &ia_vert_info);
659 
660          if (ia_vert_info.count) {
661             FREE(vert_info->verts);
662             if (free_prim_info) {
663                FREE(prim_info->primitive_lengths);
664                FREE(tes_elts_out);
665                tes_elts_out = NULL;
666             }
667             vert_info = &ia_vert_info;
668             prim_info = &ia_prim_info;
669             free_prim_info = true;
670          }
671       }
672    }
673 
674    /* stream output needs to be done before clipping */
675    draw_pt_so_emit(fpme->so_emit,
676                    gshader ? gshader->num_vertex_streams : 1,
677                    vert_info, prim_info);
678 
679    if (prim_info->count == 0) {
680       debug_printf("GS/IA didn't emit any vertices!\n");
681    } else {
682       draw_stats_clipper_primitives(draw, prim_info);
683 
684       /*
685        * if there's no position, need to stop now, or the latter stages
686        * will try to access non-existent position output.
687        */
688       if (draw_current_shader_position_output(draw) != -1) {
689          if ((opt & PT_SHADE) &&
690              (gshader || tes_shader ||
691               draw->vs.vertex_shader->info.writes_viewport_index)) {
692             clipped = draw_pt_post_vs_run(fpme->post_vs, vert_info, prim_info);
693          }
694          /* "clipped" also includes non-one edgeflag */
695          if (clipped) {
696             opt |= PT_PIPELINE;
697          }
698 
699          /* Do we need to run the pipeline? Now will come here if clipped */
700          if (opt & PT_PIPELINE) {
701             pipeline(fpme, vert_info, prim_info);
702          } else {
703             emit(fpme->emit, vert_info, prim_info);
704          }
705       }
706    }
707 
708    FREE(vert_info->verts);
709    if (gshader && gshader->num_vertex_streams > 1)
710      for (unsigned i = 1; i < gshader->num_vertex_streams; i++)
711        FREE(gs_vert_info[i].verts);
712 
713    if (free_prim_info) {
714       FREE(tes_elts_out);
715       FREE(prim_info->primitive_lengths);
716    }
717 }
718 
719 
720 static inline enum mesa_prim
prim_type(enum mesa_prim prim,unsigned flags)721 prim_type(enum mesa_prim prim, unsigned flags)
722 {
723    if (flags & DRAW_LINE_LOOP_AS_STRIP)
724       return MESA_PRIM_LINE_STRIP;
725    else
726       return prim;
727 }
728 
729 
730 static void
llvm_middle_end_run(struct draw_pt_middle_end * middle,const unsigned * fetch_elts,unsigned fetch_count,const uint16_t * draw_elts,unsigned draw_count,unsigned prim_flags)731 llvm_middle_end_run(struct draw_pt_middle_end *middle,
732                     const unsigned *fetch_elts,
733                     unsigned fetch_count,
734                     const uint16_t *draw_elts,
735                     unsigned draw_count,
736                     unsigned prim_flags)
737 {
738    struct llvm_middle_end *fpme = llvm_middle_end(middle);
739    struct draw_fetch_info fetch_info;
740    struct draw_prim_info prim_info;
741 
742    fetch_info.linear = false;
743    fetch_info.start = 0;
744    fetch_info.elts = fetch_elts;
745    fetch_info.count = fetch_count;
746 
747    prim_info.linear = false;
748    prim_info.start = 0;
749    prim_info.count = draw_count;
750    prim_info.elts = draw_elts;
751    prim_info.prim = prim_type(fpme->input_prim, prim_flags);
752    prim_info.flags = prim_flags;
753    prim_info.primitive_count = 1;
754    prim_info.primitive_lengths = &draw_count;
755 
756    llvm_pipeline_generic(middle, &fetch_info, &prim_info);
757 }
758 
759 
760 static void
llvm_middle_end_linear_run(struct draw_pt_middle_end * middle,unsigned start,unsigned count,unsigned prim_flags)761 llvm_middle_end_linear_run(struct draw_pt_middle_end *middle,
762                            unsigned start,
763                            unsigned count,
764                            unsigned prim_flags)
765 {
766    struct llvm_middle_end *fpme = llvm_middle_end(middle);
767    struct draw_fetch_info fetch_info;
768    struct draw_prim_info prim_info;
769 
770    fetch_info.linear = true;
771    fetch_info.start = start;
772    fetch_info.count = count;
773    fetch_info.elts = NULL;
774 
775    prim_info.linear = true;
776    prim_info.start = start;
777    prim_info.count = count;
778    prim_info.elts = NULL;
779    prim_info.prim = prim_type(fpme->input_prim, prim_flags);
780    prim_info.flags = prim_flags;
781    prim_info.primitive_count = 1;
782    prim_info.primitive_lengths = &count;
783 
784    llvm_pipeline_generic(middle, &fetch_info, &prim_info);
785 }
786 
787 
788 static bool
llvm_middle_end_linear_run_elts(struct draw_pt_middle_end * middle,unsigned start,unsigned count,const uint16_t * draw_elts,unsigned draw_count,unsigned prim_flags)789 llvm_middle_end_linear_run_elts(struct draw_pt_middle_end *middle,
790                                 unsigned start,
791                                 unsigned count,
792                                 const uint16_t *draw_elts,
793                                 unsigned draw_count,
794                                 unsigned prim_flags)
795 {
796    struct llvm_middle_end *fpme = llvm_middle_end(middle);
797    struct draw_fetch_info fetch_info;
798    struct draw_prim_info prim_info;
799 
800    fetch_info.linear = true;
801    fetch_info.start = start;
802    fetch_info.count = count;
803    fetch_info.elts = NULL;
804 
805    prim_info.linear = false;
806    prim_info.start = 0;
807    prim_info.count = draw_count;
808    prim_info.elts = draw_elts;
809    prim_info.prim = prim_type(fpme->input_prim, prim_flags);
810    prim_info.flags = prim_flags;
811    prim_info.primitive_count = 1;
812    prim_info.primitive_lengths = &draw_count;
813 
814    llvm_pipeline_generic(middle, &fetch_info, &prim_info);
815 
816    return true;
817 }
818 
819 
820 static void
llvm_middle_end_finish(struct draw_pt_middle_end * middle)821 llvm_middle_end_finish(struct draw_pt_middle_end *middle)
822 {
823    /* nothing to do */
824 }
825 
826 
827 static void
llvm_middle_end_destroy(struct draw_pt_middle_end * middle)828 llvm_middle_end_destroy(struct draw_pt_middle_end *middle)
829 {
830    struct llvm_middle_end *fpme = llvm_middle_end(middle);
831 
832    if (fpme->fetch)
833       draw_pt_fetch_destroy(fpme->fetch);
834 
835    if (fpme->emit)
836       draw_pt_emit_destroy(fpme->emit);
837 
838    if (fpme->so_emit)
839       draw_pt_so_emit_destroy(fpme->so_emit);
840 
841    if (fpme->post_vs)
842       draw_pt_post_vs_destroy(fpme->post_vs);
843 
844    FREE(middle);
845 }
846 
847 
848 struct draw_pt_middle_end *
draw_pt_fetch_pipeline_or_emit_llvm(struct draw_context * draw)849 draw_pt_fetch_pipeline_or_emit_llvm(struct draw_context *draw)
850 {
851    struct llvm_middle_end *fpme = 0;
852 
853    if (!draw->llvm)
854       return NULL;
855 
856    fpme = CALLOC_STRUCT(llvm_middle_end);
857    if (!fpme)
858       goto fail;
859 
860    fpme->base.prepare         = llvm_middle_end_prepare;
861    fpme->base.bind_parameters = llvm_middle_end_bind_parameters;
862    fpme->base.run             = llvm_middle_end_run;
863    fpme->base.run_linear      = llvm_middle_end_linear_run;
864    fpme->base.run_linear_elts = llvm_middle_end_linear_run_elts;
865    fpme->base.finish          = llvm_middle_end_finish;
866    fpme->base.destroy         = llvm_middle_end_destroy;
867 
868    fpme->draw = draw;
869 
870    fpme->fetch = draw_pt_fetch_create(draw);
871    if (!fpme->fetch)
872       goto fail;
873 
874    fpme->post_vs = draw_pt_post_vs_create(draw);
875    if (!fpme->post_vs)
876       goto fail;
877 
878    fpme->emit = draw_pt_emit_create(draw);
879    if (!fpme->emit)
880       goto fail;
881 
882    fpme->so_emit = draw_pt_so_emit_create(draw);
883    if (!fpme->so_emit)
884       goto fail;
885 
886    fpme->llvm = draw->llvm;
887    if (!fpme->llvm)
888       goto fail;
889 
890    fpme->current_variant = NULL;
891 
892    return &fpme->base;
893 
894  fail:
895    if (fpme)
896       llvm_middle_end_destroy(&fpme->base);
897 
898    return NULL;
899 }
900