2 * Copyright 2010 Christoph Bumiller
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
18 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20 * OTHER DEALINGS IN THE SOFTWARE.
23 #include "nv50/nv50_program.h"
24 #include "nv50/nv50_context.h"
26 #include "codegen/nv50_ir_driver.h"
28 static inline unsigned
29 bitcount4(const uint32_t val
)
31 static const uint8_t cnt
[16]
32 = { 0, 1, 1, 2, 1, 2, 2, 3, 1, 2, 2, 3, 2, 3, 3, 4 };
33 return cnt
[val
& 0xf];
37 nv50_vertprog_assign_slots(struct nv50_ir_prog_info
*info
)
39 struct nv50_program
*prog
= (struct nv50_program
*)info
->driverPriv
;
43 for (i
= 0; i
< info
->numInputs
; ++i
) {
45 prog
->in
[i
].sn
= info
->in
[i
].sn
;
46 prog
->in
[i
].si
= info
->in
[i
].si
;
48 prog
->in
[i
].mask
= info
->in
[i
].mask
;
50 prog
->vp
.attrs
[(4 * i
) / 32] |= info
->in
[i
].mask
<< ((4 * i
) % 32);
52 for (c
= 0; c
< 4; ++c
)
53 if (info
->in
[i
].mask
& (1 << c
))
54 info
->in
[i
].slot
[c
] = n
++;
56 if (info
->in
[i
].sn
== TGSI_SEMANTIC_PRIMID
)
57 prog
->vp
.attrs
[2] |= NV50_3D_VP_GP_BUILTIN_ATTR_EN_PRIMITIVE_ID
;
59 prog
->in_nr
= info
->numInputs
;
61 for (i
= 0; i
< info
->numSysVals
; ++i
) {
62 switch (info
->sv
[i
].sn
) {
63 case TGSI_SEMANTIC_INSTANCEID
:
64 prog
->vp
.attrs
[2] |= NV50_3D_VP_GP_BUILTIN_ATTR_EN_INSTANCE_ID
;
66 case TGSI_SEMANTIC_VERTEXID
:
67 prog
->vp
.attrs
[2] |= NV50_3D_VP_GP_BUILTIN_ATTR_EN_VERTEX_ID
;
68 prog
->vp
.attrs
[2] |= NV50_3D_VP_GP_BUILTIN_ATTR_EN_VERTEX_ID_DRAW_ARRAYS_ADD_START
;
76 * Corner case: VP has no inputs, but we will still need to submit data to
77 * draw it. HW will shout at us and won't draw anything if we don't enable
78 * any input, so let's just pretend it's the first one.
80 if (prog
->vp
.attrs
[0] == 0 &&
81 prog
->vp
.attrs
[1] == 0 &&
82 prog
->vp
.attrs
[2] == 0)
83 prog
->vp
.attrs
[0] |= 0xf;
85 /* VertexID before InstanceID */
86 if (info
->io
.vertexId
< info
->numSysVals
)
87 info
->sv
[info
->io
.vertexId
].slot
[0] = n
++;
88 if (info
->io
.instanceId
< info
->numSysVals
)
89 info
->sv
[info
->io
.instanceId
].slot
[0] = n
++;
92 for (i
= 0; i
< info
->numOutputs
; ++i
) {
93 switch (info
->out
[i
].sn
) {
94 case TGSI_SEMANTIC_PSIZE
:
97 case TGSI_SEMANTIC_CLIPDIST
:
98 prog
->vp
.clpd
[info
->out
[i
].si
] = n
;
100 case TGSI_SEMANTIC_EDGEFLAG
:
101 prog
->vp
.edgeflag
= i
;
103 case TGSI_SEMANTIC_BCOLOR
:
104 prog
->vp
.bfc
[info
->out
[i
].si
] = i
;
106 case TGSI_SEMANTIC_LAYER
:
107 prog
->gp
.has_layer
= true;
108 prog
->gp
.layerid
= n
;
110 case TGSI_SEMANTIC_VIEWPORT_INDEX
:
111 prog
->gp
.has_viewport
= true;
112 prog
->gp
.viewportid
= n
;
118 prog
->out
[i
].sn
= info
->out
[i
].sn
;
119 prog
->out
[i
].si
= info
->out
[i
].si
;
121 prog
->out
[i
].mask
= info
->out
[i
].mask
;
123 for (c
= 0; c
< 4; ++c
)
124 if (info
->out
[i
].mask
& (1 << c
))
125 info
->out
[i
].slot
[c
] = n
++;
127 prog
->out_nr
= info
->numOutputs
;
132 if (prog
->vp
.psiz
< info
->numOutputs
)
133 prog
->vp
.psiz
= prog
->out
[prog
->vp
.psiz
].hw
;
139 nv50_fragprog_assign_slots(struct nv50_ir_prog_info
*info
)
141 struct nv50_program
*prog
= (struct nv50_program
*)info
->driverPriv
;
147 /* count recorded non-flat inputs */
148 for (m
= 0, i
= 0; i
< info
->numInputs
; ++i
) {
149 switch (info
->in
[i
].sn
) {
150 case TGSI_SEMANTIC_POSITION
:
153 m
+= info
->in
[i
].flat
? 0 : 1;
157 /* careful: id may be != i in info->in[prog->in[i].id] */
159 /* Fill prog->in[] so that non-flat inputs are first and
160 * kick out special inputs that don't use the RESULT_MAP.
162 for (n
= 0, i
= 0; i
< info
->numInputs
; ++i
) {
163 if (info
->in
[i
].sn
== TGSI_SEMANTIC_POSITION
) {
164 prog
->fp
.interp
|= info
->in
[i
].mask
<< 24;
165 for (c
= 0; c
< 4; ++c
)
166 if (info
->in
[i
].mask
& (1 << c
))
167 info
->in
[i
].slot
[c
] = nintp
++;
169 unsigned j
= info
->in
[i
].flat
? m
++ : n
++;
171 if (info
->in
[i
].sn
== TGSI_SEMANTIC_COLOR
)
172 prog
->vp
.bfc
[info
->in
[i
].si
] = j
;
173 else if (info
->in
[i
].sn
== TGSI_SEMANTIC_PRIMID
)
174 prog
->vp
.attrs
[2] |= NV50_3D_VP_GP_BUILTIN_ATTR_EN_PRIMITIVE_ID
;
177 prog
->in
[j
].mask
= info
->in
[i
].mask
;
178 prog
->in
[j
].sn
= info
->in
[i
].sn
;
179 prog
->in
[j
].si
= info
->in
[i
].si
;
180 prog
->in
[j
].linear
= info
->in
[i
].linear
;
185 if (!(prog
->fp
.interp
& (8 << 24))) {
187 prog
->fp
.interp
|= 8 << 24;
190 for (i
= 0; i
< prog
->in_nr
; ++i
) {
191 int j
= prog
->in
[i
].id
;
193 prog
->in
[i
].hw
= nintp
;
194 for (c
= 0; c
< 4; ++c
)
195 if (prog
->in
[i
].mask
& (1 << c
))
196 info
->in
[j
].slot
[c
] = nintp
++;
198 /* (n == m) if m never increased, i.e. no flat inputs */
199 nflat
= (n
< m
) ? (nintp
- prog
->in
[n
].hw
) : 0;
200 nintp
-= bitcount4(prog
->fp
.interp
>> 24); /* subtract position inputs */
201 nvary
= nintp
- nflat
;
203 prog
->fp
.interp
|= nvary
<< NV50_3D_FP_INTERPOLANT_CTRL_COUNT_NONFLAT__SHIFT
;
204 prog
->fp
.interp
|= nintp
<< NV50_3D_FP_INTERPOLANT_CTRL_COUNT__SHIFT
;
206 /* put front/back colors right after HPOS */
207 prog
->fp
.colors
= 4 << NV50_3D_SEMANTIC_COLOR_FFC0_ID__SHIFT
;
208 for (i
= 0; i
< 2; ++i
)
209 if (prog
->vp
.bfc
[i
] < 0xff)
210 prog
->fp
.colors
+= bitcount4(prog
->in
[prog
->vp
.bfc
[i
]].mask
) << 16;
214 if (info
->prop
.fp
.numColourResults
> 1)
215 prog
->fp
.flags
[0] |= NV50_3D_FP_CONTROL_MULTIPLE_RESULTS
;
217 for (i
= 0; i
< info
->numOutputs
; ++i
) {
219 prog
->out
[i
].sn
= info
->out
[i
].sn
;
220 prog
->out
[i
].si
= info
->out
[i
].si
;
221 prog
->out
[i
].mask
= info
->out
[i
].mask
;
223 if (i
== info
->io
.fragDepth
|| i
== info
->io
.sampleMask
)
225 prog
->out
[i
].hw
= info
->out
[i
].si
* 4;
227 for (c
= 0; c
< 4; ++c
)
228 info
->out
[i
].slot
[c
] = prog
->out
[i
].hw
+ c
;
230 prog
->max_out
= MAX2(prog
->max_out
, prog
->out
[i
].hw
+ 4);
233 if (info
->io
.sampleMask
< PIPE_MAX_SHADER_OUTPUTS
) {
234 info
->out
[info
->io
.sampleMask
].slot
[0] = prog
->max_out
++;
235 prog
->fp
.has_samplemask
= 1;
238 if (info
->io
.fragDepth
< PIPE_MAX_SHADER_OUTPUTS
)
239 info
->out
[info
->io
.fragDepth
].slot
[2] = prog
->max_out
++;
248 nv50_program_assign_varying_slots(struct nv50_ir_prog_info
*info
)
250 switch (info
->type
) {
251 case PIPE_SHADER_VERTEX
:
252 return nv50_vertprog_assign_slots(info
);
253 case PIPE_SHADER_GEOMETRY
:
254 return nv50_vertprog_assign_slots(info
);
255 case PIPE_SHADER_FRAGMENT
:
256 return nv50_fragprog_assign_slots(info
);
257 case PIPE_SHADER_COMPUTE
:
264 static struct nv50_stream_output_state
*
265 nv50_program_create_strmout_state(const struct nv50_ir_prog_info
*info
,
266 const struct pipe_stream_output_info
*pso
)
268 struct nv50_stream_output_state
*so
;
272 so
= MALLOC_STRUCT(nv50_stream_output_state
);
275 memset(so
->map
, 0xff, sizeof(so
->map
));
277 for (b
= 0; b
< 4; ++b
)
278 so
->num_attribs
[b
] = 0;
279 for (i
= 0; i
< pso
->num_outputs
; ++i
) {
280 unsigned end
= pso
->output
[i
].dst_offset
+ pso
->output
[i
].num_components
;
281 b
= pso
->output
[i
].output_buffer
;
283 so
->num_attribs
[b
] = MAX2(so
->num_attribs
[b
], end
);
286 so
->ctrl
= NV50_3D_STRMOUT_BUFFERS_CTRL_INTERLEAVED
;
288 so
->stride
[0] = pso
->stride
[0] * 4;
290 for (b
= 1; b
< 4; ++b
) {
291 assert(!so
->num_attribs
[b
] || so
->num_attribs
[b
] == pso
->stride
[b
]);
292 so
->stride
[b
] = so
->num_attribs
[b
] * 4;
293 if (so
->num_attribs
[b
])
294 so
->ctrl
= (b
+ 1) << NV50_3D_STRMOUT_BUFFERS_CTRL_SEPARATE__SHIFT
;
295 base
[b
] = align(base
[b
- 1] + so
->num_attribs
[b
- 1], 4);
297 if (so
->ctrl
& NV50_3D_STRMOUT_BUFFERS_CTRL_INTERLEAVED
) {
298 assert(so
->stride
[0] < NV50_3D_STRMOUT_BUFFERS_CTRL_STRIDE__MAX
);
299 so
->ctrl
|= so
->stride
[0] << NV50_3D_STRMOUT_BUFFERS_CTRL_STRIDE__SHIFT
;
302 so
->map_size
= base
[3] + so
->num_attribs
[3];
304 for (i
= 0; i
< pso
->num_outputs
; ++i
) {
305 const unsigned s
= pso
->output
[i
].start_component
;
306 const unsigned p
= pso
->output
[i
].dst_offset
;
307 const unsigned r
= pso
->output
[i
].register_index
;
308 b
= pso
->output
[i
].output_buffer
;
310 for (c
= 0; c
< pso
->output
[i
].num_components
; ++c
)
311 so
->map
[base
[b
] + p
+ c
] = info
->out
[r
].slot
[s
+ c
];
318 nv50_program_translate(struct nv50_program
*prog
, uint16_t chipset
,
319 struct pipe_debug_callback
*debug
)
321 struct nv50_ir_prog_info
*info
;
323 const uint8_t map_undef
= (prog
->type
== PIPE_SHADER_VERTEX
) ? 0x40 : 0x80;
325 info
= CALLOC_STRUCT(nv50_ir_prog_info
);
329 info
->type
= prog
->type
;
330 info
->target
= chipset
;
331 info
->bin
.sourceRep
= NV50_PROGRAM_IR_TGSI
;
332 info
->bin
.source
= (void *)prog
->pipe
.tokens
;
334 info
->io
.auxCBSlot
= 15;
335 info
->io
.ucpBase
= NV50_CB_AUX_UCP_OFFSET
;
336 info
->io
.genUserClip
= prog
->vp
.clpd_nr
;
338 info
->io
.suInfoBase
= NV50_CB_AUX_TEX_MS_OFFSET
;
339 info
->io
.sampleInfoBase
= NV50_CB_AUX_SAMPLE_OFFSET
;
340 info
->io
.msInfoCBSlot
= 15;
341 info
->io
.msInfoBase
= NV50_CB_AUX_MS_OFFSET
;
343 info
->assignSlots
= nv50_program_assign_varying_slots
;
345 prog
->vp
.bfc
[0] = 0xff;
346 prog
->vp
.bfc
[1] = 0xff;
347 prog
->vp
.edgeflag
= 0xff;
348 prog
->vp
.clpd
[0] = map_undef
;
349 prog
->vp
.clpd
[1] = map_undef
;
350 prog
->vp
.psiz
= map_undef
;
351 prog
->gp
.has_layer
= 0;
352 prog
->gp
.has_viewport
= 0;
354 if (prog
->type
== PIPE_SHADER_COMPUTE
)
355 info
->prop
.cp
.inputOffset
= 0x10;
357 info
->driverPriv
= prog
;
360 info
->optLevel
= debug_get_num_option("NV50_PROG_OPTIMIZE", 3);
361 info
->dbgFlags
= debug_get_num_option("NV50_PROG_DEBUG", 0);
366 ret
= nv50_ir_generate_code(info
);
368 NOUVEAU_ERR("shader translation failed: %i\n", ret
);
372 prog
->code
= info
->bin
.code
;
373 prog
->code_size
= info
->bin
.codeSize
;
374 prog
->fixups
= info
->bin
.relocData
;
375 prog
->interps
= info
->bin
.fixupData
;
376 prog
->max_gpr
= MAX2(4, (info
->bin
.maxGPR
>> 1) + 1);
377 prog
->tls_space
= info
->bin
.tlsSpace
;
379 prog
->vp
.need_vertex_id
= info
->io
.vertexId
< PIPE_MAX_SHADER_INPUTS
;
381 prog
->vp
.clip_enable
= (1 << info
->io
.clipDistances
) - 1;
382 prog
->vp
.cull_enable
=
383 ((1 << info
->io
.cullDistances
) - 1) << info
->io
.clipDistances
;
384 prog
->vp
.clip_mode
= 0;
385 for (i
= 0; i
< info
->io
.cullDistances
; ++i
)
386 prog
->vp
.clip_mode
|= 1 << ((info
->io
.clipDistances
+ i
) * 4);
388 if (prog
->type
== PIPE_SHADER_FRAGMENT
) {
389 if (info
->prop
.fp
.writesDepth
) {
390 prog
->fp
.flags
[0] |= NV50_3D_FP_CONTROL_EXPORTS_Z
;
391 prog
->fp
.flags
[1] = 0x11;
393 if (info
->prop
.fp
.usesDiscard
)
394 prog
->fp
.flags
[0] |= NV50_3D_FP_CONTROL_USES_KIL
;
396 if (prog
->type
== PIPE_SHADER_GEOMETRY
) {
397 switch (info
->prop
.gp
.outputPrim
) {
398 case PIPE_PRIM_LINE_STRIP
:
399 prog
->gp
.prim_type
= NV50_3D_GP_OUTPUT_PRIMITIVE_TYPE_LINE_STRIP
;
401 case PIPE_PRIM_TRIANGLE_STRIP
:
402 prog
->gp
.prim_type
= NV50_3D_GP_OUTPUT_PRIMITIVE_TYPE_TRIANGLE_STRIP
;
404 case PIPE_PRIM_POINTS
:
406 assert(info
->prop
.gp
.outputPrim
== PIPE_PRIM_POINTS
);
407 prog
->gp
.prim_type
= NV50_3D_GP_OUTPUT_PRIMITIVE_TYPE_POINTS
;
410 prog
->gp
.vert_count
= info
->prop
.gp
.maxVertices
;
413 if (prog
->type
== PIPE_SHADER_COMPUTE
) {
414 prog
->cp
.syms
= info
->bin
.syms
;
415 prog
->cp
.num_syms
= info
->bin
.numSyms
;
417 FREE(info
->bin
.syms
);
420 if (prog
->pipe
.stream_output
.num_outputs
)
421 prog
->so
= nv50_program_create_strmout_state(info
,
422 &prog
->pipe
.stream_output
);
424 pipe_debug_message(debug
, SHADER_INFO
,
425 "type: %d, local: %d, gpr: %d, inst: %d, bytes: %d",
426 prog
->type
, info
->bin
.tlsSpace
, prog
->max_gpr
,
427 info
->bin
.instructions
, info
->bin
.codeSize
);
435 nv50_program_upload_code(struct nv50_context
*nv50
, struct nv50_program
*prog
)
437 struct nouveau_heap
*heap
;
439 uint32_t size
= align(prog
->code_size
, 0x40);
442 switch (prog
->type
) {
443 case PIPE_SHADER_VERTEX
: heap
= nv50
->screen
->vp_code_heap
; break;
444 case PIPE_SHADER_GEOMETRY
: heap
= nv50
->screen
->gp_code_heap
; break;
445 case PIPE_SHADER_FRAGMENT
: heap
= nv50
->screen
->fp_code_heap
; break;
446 case PIPE_SHADER_COMPUTE
: heap
= nv50
->screen
->fp_code_heap
; break;
448 assert(!"invalid program type");
452 ret
= nouveau_heap_alloc(heap
, size
, prog
, &prog
->mem
);
454 /* Out of space: evict everything to compactify the code segment, hoping
455 * the working set is much smaller and drifts slowly. Improve me !
458 struct nv50_program
*evict
= heap
->next
->priv
;
460 nouveau_heap_free(&evict
->mem
);
462 debug_printf("WARNING: out of code space, evicting all shaders.\n");
463 ret
= nouveau_heap_alloc(heap
, size
, prog
, &prog
->mem
);
465 NOUVEAU_ERR("shader too large (0x%x) to fit in code space ?\n", size
);
470 if (prog
->type
== PIPE_SHADER_COMPUTE
) {
471 /* CP code must be uploaded in FP code segment. */
474 prog
->code_base
= prog
->mem
->start
;
475 prog_type
= prog
->type
;
478 ret
= nv50_tls_realloc(nv50
->screen
, prog
->tls_space
);
480 nouveau_heap_free(&prog
->mem
);
484 nv50
->state
.new_tls_space
= true;
487 nv50_ir_relocate_code(prog
->fixups
, prog
->code
, prog
->code_base
, 0, 0);
489 nv50_ir_apply_fixups(prog
->interps
, prog
->code
,
490 prog
->fp
.force_persample_interp
,
491 false /* flatshade */);
493 nv50_sifc_linear_u8(&nv50
->base
, nv50
->screen
->code
,
494 (prog_type
<< NV50_CODE_BO_SIZE_LOG2
) + prog
->code_base
,
495 NOUVEAU_BO_VRAM
, prog
->code_size
, prog
->code
);
497 BEGIN_NV04(nv50
->base
.pushbuf
, NV50_3D(CODE_CB_FLUSH
), 1);
498 PUSH_DATA (nv50
->base
.pushbuf
, 0);
504 nv50_program_destroy(struct nv50_context
*nv50
, struct nv50_program
*p
)
506 const struct pipe_shader_state pipe
= p
->pipe
;
507 const ubyte type
= p
->type
;
510 nouveau_heap_free(&p
->mem
);
518 if (type
== PIPE_SHADER_COMPUTE
)
521 memset(p
, 0, sizeof(*p
));