2 * Copyright 2011 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 "codegen/nv50_ir.h"
24 #include "codegen/nv50_ir_target.h"
25 #include "codegen/nv50_ir_build_util.h"
28 #include "util/u_math.h"
34 Instruction::isNop() const
36 if (op
== OP_PHI
|| op
== OP_SPLIT
|| op
== OP_MERGE
|| op
== OP_CONSTRAINT
)
38 if (terminator
|| join
) // XXX: should terminator imply flow ?
42 if (!fixed
&& op
== OP_NOP
)
45 if (defExists(0) && def(0).rep()->reg
.data
.id
< 0) {
46 for (int d
= 1; defExists(d
); ++d
)
47 if (def(d
).rep()->reg
.data
.id
>= 0)
48 WARN("part of vector result is unused !\n");
52 if (op
== OP_MOV
|| op
== OP_UNION
) {
53 if (!getDef(0)->equals(getSrc(0)))
56 if (!def(0).rep()->equals(getSrc(1)))
64 bool Instruction::isDead() const
69 op
== OP_SUSTB
|| op
== OP_SUSTP
|| op
== OP_SUREDP
|| op
== OP_SUREDB
||
73 for (int d
= 0; defExists(d
); ++d
)
74 if (getDef(d
)->refCount() || getDef(d
)->reg
.data
.id
>= 0)
77 if (terminator
|| asFlow())
85 // =============================================================================
87 class CopyPropagation
: public Pass
90 virtual bool visit(BasicBlock
*);
93 // Propagate all MOVs forward to make subsequent optimization easier, except if
94 // the sources stem from a phi, in which case we don't want to mess up potential
95 // swaps $rX <-> $rY, i.e. do not create live range overlaps of phi src and def.
97 CopyPropagation::visit(BasicBlock
*bb
)
99 Instruction
*mov
, *si
, *next
;
101 for (mov
= bb
->getEntry(); mov
; mov
= next
) {
103 if (mov
->op
!= OP_MOV
|| mov
->fixed
|| !mov
->getSrc(0)->asLValue())
105 if (mov
->getPredicate())
107 if (mov
->def(0).getFile() != mov
->src(0).getFile())
109 si
= mov
->getSrc(0)->getInsn();
110 if (mov
->getDef(0)->reg
.data
.id
< 0 && si
&& si
->op
!= OP_PHI
) {
112 mov
->def(0).replace(mov
->getSrc(0), false);
113 delete_Instruction(prog
, mov
);
119 // =============================================================================
121 class MergeSplits
: public Pass
124 virtual bool visit(BasicBlock
*);
127 // For SPLIT / MERGE pairs that operate on the same registers, replace the
128 // post-merge def with the SPLIT's source.
130 MergeSplits::visit(BasicBlock
*bb
)
132 Instruction
*i
, *next
, *si
;
134 for (i
= bb
->getEntry(); i
; i
= next
) {
136 if (i
->op
!= OP_MERGE
|| typeSizeof(i
->dType
) != 8)
138 si
= i
->getSrc(0)->getInsn();
139 if (si
->op
!= OP_SPLIT
|| si
!= i
->getSrc(1)->getInsn())
141 i
->def(0).replace(si
->getSrc(0), false);
142 delete_Instruction(prog
, i
);
148 // =============================================================================
150 class LoadPropagation
: public Pass
153 virtual bool visit(BasicBlock
*);
155 void checkSwapSrc01(Instruction
*);
157 bool isCSpaceLoad(Instruction
*);
158 bool isImmdLoad(Instruction
*);
159 bool isAttribOrSharedLoad(Instruction
*);
163 LoadPropagation::isCSpaceLoad(Instruction
*ld
)
165 return ld
&& ld
->op
== OP_LOAD
&& ld
->src(0).getFile() == FILE_MEMORY_CONST
;
169 LoadPropagation::isImmdLoad(Instruction
*ld
)
171 if (!ld
|| (ld
->op
!= OP_MOV
) ||
172 ((typeSizeof(ld
->dType
) != 4) && (typeSizeof(ld
->dType
) != 8)))
175 // A 0 can be replaced with a register, so it doesn't count as an immediate.
177 return ld
->src(0).getImmediate(val
) && !val
.isInteger(0);
181 LoadPropagation::isAttribOrSharedLoad(Instruction
*ld
)
184 (ld
->op
== OP_VFETCH
||
185 (ld
->op
== OP_LOAD
&&
186 (ld
->src(0).getFile() == FILE_SHADER_INPUT
||
187 ld
->src(0).getFile() == FILE_MEMORY_SHARED
)));
191 LoadPropagation::checkSwapSrc01(Instruction
*insn
)
193 const Target
*targ
= prog
->getTarget();
194 if (!targ
->getOpInfo(insn
).commutative
)
195 if (insn
->op
!= OP_SET
&& insn
->op
!= OP_SLCT
&& insn
->op
!= OP_SUB
)
197 if (insn
->src(1).getFile() != FILE_GPR
)
199 // This is the special OP_SET used for alphatesting, we can't reverse its
200 // arguments as that will confuse the fixup code.
201 if (insn
->op
== OP_SET
&& insn
->subOp
)
204 Instruction
*i0
= insn
->getSrc(0)->getInsn();
205 Instruction
*i1
= insn
->getSrc(1)->getInsn();
207 // Swap sources to inline the less frequently used source. That way,
208 // optimistically, it will eventually be able to remove the instruction.
209 int i0refs
= insn
->getSrc(0)->refCount();
210 int i1refs
= insn
->getSrc(1)->refCount();
212 if ((isCSpaceLoad(i0
) || isImmdLoad(i0
)) && targ
->insnCanLoad(insn
, 1, i0
)) {
213 if ((!isImmdLoad(i1
) && !isCSpaceLoad(i1
)) ||
214 !targ
->insnCanLoad(insn
, 1, i1
) ||
216 insn
->swapSources(0, 1);
220 if (isAttribOrSharedLoad(i1
)) {
221 if (!isAttribOrSharedLoad(i0
))
222 insn
->swapSources(0, 1);
229 if (insn
->op
== OP_SET
|| insn
->op
== OP_SET_AND
||
230 insn
->op
== OP_SET_OR
|| insn
->op
== OP_SET_XOR
)
231 insn
->asCmp()->setCond
= reverseCondCode(insn
->asCmp()->setCond
);
233 if (insn
->op
== OP_SLCT
)
234 insn
->asCmp()->setCond
= inverseCondCode(insn
->asCmp()->setCond
);
236 if (insn
->op
== OP_SUB
) {
237 insn
->src(0).mod
= insn
->src(0).mod
^ Modifier(NV50_IR_MOD_NEG
);
238 insn
->src(1).mod
= insn
->src(1).mod
^ Modifier(NV50_IR_MOD_NEG
);
243 LoadPropagation::visit(BasicBlock
*bb
)
245 const Target
*targ
= prog
->getTarget();
248 for (Instruction
*i
= bb
->getEntry(); i
; i
= next
) {
251 if (i
->op
== OP_CALL
) // calls have args as sources, they must be in regs
254 if (i
->op
== OP_PFETCH
) // pfetch expects arg1 to be a reg
260 for (int s
= 0; i
->srcExists(s
); ++s
) {
261 Instruction
*ld
= i
->getSrc(s
)->getInsn();
263 if (!ld
|| ld
->fixed
|| (ld
->op
!= OP_LOAD
&& ld
->op
!= OP_MOV
))
265 if (!targ
->insnCanLoad(i
, s
, ld
))
269 i
->setSrc(s
, ld
->getSrc(0));
270 if (ld
->src(0).isIndirect(0))
271 i
->setIndirect(s
, 0, ld
->getIndirect(0, 0));
273 if (ld
->getDef(0)->refCount() == 0)
274 delete_Instruction(prog
, ld
);
280 // =============================================================================
282 class IndirectPropagation
: public Pass
285 virtual bool visit(BasicBlock
*);
289 IndirectPropagation::visit(BasicBlock
*bb
)
291 const Target
*targ
= prog
->getTarget();
294 for (Instruction
*i
= bb
->getEntry(); i
; i
= next
) {
297 for (int s
= 0; i
->srcExists(s
); ++s
) {
300 if (!i
->src(s
).isIndirect(0))
302 insn
= i
->getIndirect(s
, 0)->getInsn();
305 if (insn
->op
== OP_ADD
&& !isFloatType(insn
->dType
)) {
306 if (insn
->src(0).getFile() != targ
->nativeFile(FILE_ADDRESS
) ||
307 !insn
->src(1).getImmediate(imm
) ||
308 !targ
->insnCanLoadOffset(i
, s
, imm
.reg
.data
.s32
))
310 i
->setIndirect(s
, 0, insn
->getSrc(0));
311 i
->setSrc(s
, cloneShallow(func
, i
->getSrc(s
)));
312 i
->src(s
).get()->reg
.data
.offset
+= imm
.reg
.data
.u32
;
313 } else if (insn
->op
== OP_SUB
&& !isFloatType(insn
->dType
)) {
314 if (insn
->src(0).getFile() != targ
->nativeFile(FILE_ADDRESS
) ||
315 !insn
->src(1).getImmediate(imm
) ||
316 !targ
->insnCanLoadOffset(i
, s
, -imm
.reg
.data
.s32
))
318 i
->setIndirect(s
, 0, insn
->getSrc(0));
319 i
->setSrc(s
, cloneShallow(func
, i
->getSrc(s
)));
320 i
->src(s
).get()->reg
.data
.offset
-= imm
.reg
.data
.u32
;
321 } else if (insn
->op
== OP_MOV
) {
322 if (!insn
->src(0).getImmediate(imm
) ||
323 !targ
->insnCanLoadOffset(i
, s
, imm
.reg
.data
.s32
))
325 i
->setIndirect(s
, 0, NULL
);
326 i
->setSrc(s
, cloneShallow(func
, i
->getSrc(s
)));
327 i
->src(s
).get()->reg
.data
.offset
+= imm
.reg
.data
.u32
;
334 // =============================================================================
336 // Evaluate constant expressions.
337 class ConstantFolding
: public Pass
340 bool foldAll(Program
*);
343 virtual bool visit(BasicBlock
*);
345 void expr(Instruction
*, ImmediateValue
&, ImmediateValue
&);
346 void expr(Instruction
*, ImmediateValue
&, ImmediateValue
&, ImmediateValue
&);
347 void opnd(Instruction
*, ImmediateValue
&, int s
);
348 void opnd3(Instruction
*, ImmediateValue
&);
350 void unary(Instruction
*, const ImmediateValue
&);
352 void tryCollapseChainedMULs(Instruction
*, const int s
, ImmediateValue
&);
354 CmpInstruction
*findOriginForTestWithZero(Value
*);
356 unsigned int foldCount
;
361 // TODO: remember generated immediates and only revisit these
363 ConstantFolding::foldAll(Program
*prog
)
365 unsigned int iterCount
= 0;
370 } while (foldCount
&& ++iterCount
< 2);
375 ConstantFolding::visit(BasicBlock
*bb
)
377 Instruction
*i
, *next
;
379 for (i
= bb
->getEntry(); i
; i
= next
) {
381 if (i
->op
== OP_MOV
|| i
->op
== OP_CALL
)
384 ImmediateValue src0
, src1
, src2
;
386 if (i
->srcExists(2) &&
387 i
->src(0).getImmediate(src0
) &&
388 i
->src(1).getImmediate(src1
) &&
389 i
->src(2).getImmediate(src2
))
390 expr(i
, src0
, src1
, src2
);
392 if (i
->srcExists(1) &&
393 i
->src(0).getImmediate(src0
) && i
->src(1).getImmediate(src1
))
396 if (i
->srcExists(0) && i
->src(0).getImmediate(src0
))
399 if (i
->srcExists(1) && i
->src(1).getImmediate(src1
))
401 if (i
->srcExists(2) && i
->src(2).getImmediate(src2
))
408 ConstantFolding::findOriginForTestWithZero(Value
*value
)
412 Instruction
*insn
= value
->getInsn();
416 if (insn
->asCmp() && insn
->op
!= OP_SLCT
)
417 return insn
->asCmp();
419 /* Sometimes mov's will sneak in as a result of other folding. This gets
422 if (insn
->op
== OP_MOV
)
423 return findOriginForTestWithZero(insn
->getSrc(0));
425 /* Deal with AND 1.0 here since nv50 can't fold into boolean float */
426 if (insn
->op
== OP_AND
) {
429 if (!insn
->src(s
).getImmediate(imm
)) {
431 if (!insn
->src(s
).getImmediate(imm
))
434 if (imm
.reg
.data
.f32
!= 1.0f
)
436 /* TODO: Come up with a way to handle the condition being inverted */
437 if (insn
->src(!s
).mod
!= Modifier(0))
439 return findOriginForTestWithZero(insn
->getSrc(!s
));
446 Modifier::applyTo(ImmediateValue
& imm
) const
448 if (!bits
) // avoid failure if imm.reg.type is unhandled (e.g. b128)
450 switch (imm
.reg
.type
) {
452 if (bits
& NV50_IR_MOD_ABS
)
453 imm
.reg
.data
.f32
= fabsf(imm
.reg
.data
.f32
);
454 if (bits
& NV50_IR_MOD_NEG
)
455 imm
.reg
.data
.f32
= -imm
.reg
.data
.f32
;
456 if (bits
& NV50_IR_MOD_SAT
) {
457 if (imm
.reg
.data
.f32
< 0.0f
)
458 imm
.reg
.data
.f32
= 0.0f
;
460 if (imm
.reg
.data
.f32
> 1.0f
)
461 imm
.reg
.data
.f32
= 1.0f
;
463 assert(!(bits
& NV50_IR_MOD_NOT
));
466 case TYPE_S8
: // NOTE: will be extended
469 case TYPE_U8
: // NOTE: treated as signed
472 if (bits
& NV50_IR_MOD_ABS
)
473 imm
.reg
.data
.s32
= (imm
.reg
.data
.s32
>= 0) ?
474 imm
.reg
.data
.s32
: -imm
.reg
.data
.s32
;
475 if (bits
& NV50_IR_MOD_NEG
)
476 imm
.reg
.data
.s32
= -imm
.reg
.data
.s32
;
477 if (bits
& NV50_IR_MOD_NOT
)
478 imm
.reg
.data
.s32
= ~imm
.reg
.data
.s32
;
482 if (bits
& NV50_IR_MOD_ABS
)
483 imm
.reg
.data
.f64
= fabs(imm
.reg
.data
.f64
);
484 if (bits
& NV50_IR_MOD_NEG
)
485 imm
.reg
.data
.f64
= -imm
.reg
.data
.f64
;
486 if (bits
& NV50_IR_MOD_SAT
) {
487 if (imm
.reg
.data
.f64
< 0.0)
488 imm
.reg
.data
.f64
= 0.0;
490 if (imm
.reg
.data
.f64
> 1.0)
491 imm
.reg
.data
.f64
= 1.0;
493 assert(!(bits
& NV50_IR_MOD_NOT
));
497 assert(!"invalid/unhandled type");
498 imm
.reg
.data
.u64
= 0;
504 Modifier::getOp() const
507 case NV50_IR_MOD_ABS
: return OP_ABS
;
508 case NV50_IR_MOD_NEG
: return OP_NEG
;
509 case NV50_IR_MOD_SAT
: return OP_SAT
;
510 case NV50_IR_MOD_NOT
: return OP_NOT
;
519 ConstantFolding::expr(Instruction
*i
,
520 ImmediateValue
&imm0
, ImmediateValue
&imm1
)
522 struct Storage
*const a
= &imm0
.reg
, *const b
= &imm1
.reg
;
524 DataType type
= i
->dType
;
526 memset(&res
.data
, 0, sizeof(res
.data
));
532 if (i
->dnz
&& i
->dType
== TYPE_F32
) {
533 if (!isfinite(a
->data
.f32
))
535 if (!isfinite(b
->data
.f32
))
540 res
.data
.f32
= a
->data
.f32
* b
->data
.f32
* exp2f(i
->postFactor
);
542 case TYPE_F64
: res
.data
.f64
= a
->data
.f64
* b
->data
.f64
; break;
544 if (i
->subOp
== NV50_IR_SUBOP_MUL_HIGH
) {
545 res
.data
.s32
= ((int64_t)a
->data
.s32
* b
->data
.s32
) >> 32;
550 if (i
->subOp
== NV50_IR_SUBOP_MUL_HIGH
) {
551 res
.data
.u32
= ((uint64_t)a
->data
.u32
* b
->data
.u32
) >> 32;
554 res
.data
.u32
= a
->data
.u32
* b
->data
.u32
; break;
560 if (b
->data
.u32
== 0)
563 case TYPE_F32
: res
.data
.f32
= a
->data
.f32
/ b
->data
.f32
; break;
564 case TYPE_F64
: res
.data
.f64
= a
->data
.f64
/ b
->data
.f64
; break;
565 case TYPE_S32
: res
.data
.s32
= a
->data
.s32
/ b
->data
.s32
; break;
566 case TYPE_U32
: res
.data
.u32
= a
->data
.u32
/ b
->data
.u32
; break;
573 case TYPE_F32
: res
.data
.f32
= a
->data
.f32
+ b
->data
.f32
; break;
574 case TYPE_F64
: res
.data
.f64
= a
->data
.f64
+ b
->data
.f64
; break;
576 case TYPE_U32
: res
.data
.u32
= a
->data
.u32
+ b
->data
.u32
; break;
583 case TYPE_F32
: res
.data
.f32
= a
->data
.f32
- b
->data
.f32
; break;
584 case TYPE_F64
: res
.data
.f64
= a
->data
.f64
- b
->data
.f64
; break;
586 case TYPE_U32
: res
.data
.u32
= a
->data
.u32
- b
->data
.u32
; break;
593 case TYPE_F32
: res
.data
.f32
= pow(a
->data
.f32
, b
->data
.f32
); break;
594 case TYPE_F64
: res
.data
.f64
= pow(a
->data
.f64
, b
->data
.f64
); break;
601 case TYPE_F32
: res
.data
.f32
= MAX2(a
->data
.f32
, b
->data
.f32
); break;
602 case TYPE_F64
: res
.data
.f64
= MAX2(a
->data
.f64
, b
->data
.f64
); break;
603 case TYPE_S32
: res
.data
.s32
= MAX2(a
->data
.s32
, b
->data
.s32
); break;
604 case TYPE_U32
: res
.data
.u32
= MAX2(a
->data
.u32
, b
->data
.u32
); break;
611 case TYPE_F32
: res
.data
.f32
= MIN2(a
->data
.f32
, b
->data
.f32
); break;
612 case TYPE_F64
: res
.data
.f64
= MIN2(a
->data
.f64
, b
->data
.f64
); break;
613 case TYPE_S32
: res
.data
.s32
= MIN2(a
->data
.s32
, b
->data
.s32
); break;
614 case TYPE_U32
: res
.data
.u32
= MIN2(a
->data
.u32
, b
->data
.u32
); break;
620 res
.data
.u64
= a
->data
.u64
& b
->data
.u64
;
623 res
.data
.u64
= a
->data
.u64
| b
->data
.u64
;
626 res
.data
.u64
= a
->data
.u64
^ b
->data
.u64
;
629 res
.data
.u32
= a
->data
.u32
<< b
->data
.u32
;
633 case TYPE_S32
: res
.data
.s32
= a
->data
.s32
>> b
->data
.u32
; break;
634 case TYPE_U32
: res
.data
.u32
= a
->data
.u32
>> b
->data
.u32
; break;
640 if (a
->data
.u32
!= b
->data
.u32
)
642 res
.data
.u32
= a
->data
.u32
;
645 int offset
= b
->data
.u32
& 0xff;
646 int width
= (b
->data
.u32
>> 8) & 0xff;
653 if (width
+ offset
< 32) {
655 lshift
= 32 - width
- offset
;
657 if (i
->subOp
== NV50_IR_SUBOP_EXTBF_REV
)
658 res
.data
.u32
= util_bitreverse(a
->data
.u32
);
660 res
.data
.u32
= a
->data
.u32
;
662 case TYPE_S32
: res
.data
.s32
= (res
.data
.s32
<< lshift
) >> rshift
; break;
663 case TYPE_U32
: res
.data
.u32
= (res
.data
.u32
<< lshift
) >> rshift
; break;
670 res
.data
.u32
= util_bitcount(a
->data
.u32
& b
->data
.u32
);
673 // The two arguments to pfetch are logically added together. Normally
674 // the second argument will not be constant, but that can happen.
675 res
.data
.u32
= a
->data
.u32
+ b
->data
.u32
;
683 res
.data
.u64
= (((uint64_t)b
->data
.u32
) << 32) | a
->data
.u32
;
694 i
->src(0).mod
= Modifier(0);
695 i
->src(1).mod
= Modifier(0);
698 i
->setSrc(0, new_ImmediateValue(i
->bb
->getProgram(), res
.data
.u32
));
701 i
->getSrc(0)->reg
.data
= res
.data
;
702 i
->getSrc(0)->reg
.type
= type
;
703 i
->getSrc(0)->reg
.size
= typeSizeof(type
);
708 ImmediateValue src0
, src1
= *i
->getSrc(0)->asImm();
710 // Move the immediate into position 1, where we know it might be
711 // emittable. However it might not be anyways, as there may be other
712 // restrictions, so move it into a separate LValue.
713 bld
.setPosition(i
, false);
715 i
->setSrc(1, bld
.mkMov(bld
.getSSA(type
), i
->getSrc(0), type
)->getDef(0));
716 i
->setSrc(0, i
->getSrc(2));
717 i
->src(0).mod
= i
->src(2).mod
;
720 if (i
->src(0).getImmediate(src0
))
727 // Leave PFETCH alone... we just folded its 2 args into 1.
730 i
->op
= i
->saturate
? OP_SAT
: OP_MOV
; /* SAT handled by unary() */
737 ConstantFolding::expr(Instruction
*i
,
738 ImmediateValue
&imm0
,
739 ImmediateValue
&imm1
,
740 ImmediateValue
&imm2
)
742 struct Storage
*const a
= &imm0
.reg
, *const b
= &imm1
.reg
, *const c
= &imm2
.reg
;
745 memset(&res
.data
, 0, sizeof(res
.data
));
749 int offset
= b
->data
.u32
& 0xff;
750 int width
= (b
->data
.u32
>> 8) & 0xff;
751 unsigned bitmask
= ((1 << width
) - 1) << offset
;
752 res
.data
.u32
= ((a
->data
.u32
<< offset
) & bitmask
) | (c
->data
.u32
& ~bitmask
);
759 res
.data
.f32
= a
->data
.f32
* b
->data
.f32
* exp2f(i
->postFactor
) +
763 res
.data
.f64
= a
->data
.f64
* b
->data
.f64
+ c
->data
.f64
;
766 if (i
->subOp
== NV50_IR_SUBOP_MUL_HIGH
) {
767 res
.data
.s32
= ((int64_t)a
->data
.s32
* b
->data
.s32
>> 32) + c
->data
.s32
;
772 if (i
->subOp
== NV50_IR_SUBOP_MUL_HIGH
) {
773 res
.data
.u32
= ((uint64_t)a
->data
.u32
* b
->data
.u32
>> 32) + c
->data
.u32
;
776 res
.data
.u32
= a
->data
.u32
* b
->data
.u32
+ c
->data
.u32
;
784 res
.data
.u32
= (a
->data
.u32
<< b
->data
.u32
) + c
->data
.u32
;
791 i
->src(0).mod
= Modifier(0);
792 i
->src(1).mod
= Modifier(0);
793 i
->src(2).mod
= Modifier(0);
795 i
->setSrc(0, new_ImmediateValue(i
->bb
->getProgram(), res
.data
.u32
));
799 i
->getSrc(0)->reg
.data
= res
.data
;
800 i
->getSrc(0)->reg
.type
= i
->dType
;
801 i
->getSrc(0)->reg
.size
= typeSizeof(i
->dType
);
807 ConstantFolding::unary(Instruction
*i
, const ImmediateValue
&imm
)
811 if (i
->dType
!= TYPE_F32
)
814 case OP_NEG
: res
.data
.f32
= -imm
.reg
.data
.f32
; break;
815 case OP_ABS
: res
.data
.f32
= fabsf(imm
.reg
.data
.f32
); break;
816 case OP_SAT
: res
.data
.f32
= CLAMP(imm
.reg
.data
.f32
, 0.0f
, 1.0f
); break;
817 case OP_RCP
: res
.data
.f32
= 1.0f
/ imm
.reg
.data
.f32
; break;
818 case OP_RSQ
: res
.data
.f32
= 1.0f
/ sqrtf(imm
.reg
.data
.f32
); break;
819 case OP_LG2
: res
.data
.f32
= log2f(imm
.reg
.data
.f32
); break;
820 case OP_EX2
: res
.data
.f32
= exp2f(imm
.reg
.data
.f32
); break;
821 case OP_SIN
: res
.data
.f32
= sinf(imm
.reg
.data
.f32
); break;
822 case OP_COS
: res
.data
.f32
= cosf(imm
.reg
.data
.f32
); break;
823 case OP_SQRT
: res
.data
.f32
= sqrtf(imm
.reg
.data
.f32
); break;
826 // these should be handled in subsequent OP_SIN/COS/EX2
827 res
.data
.f32
= imm
.reg
.data
.f32
;
833 i
->setSrc(0, new_ImmediateValue(i
->bb
->getProgram(), res
.data
.f32
));
834 i
->src(0).mod
= Modifier(0);
838 ConstantFolding::tryCollapseChainedMULs(Instruction
*mul2
,
839 const int s
, ImmediateValue
& imm2
)
841 const int t
= s
? 0 : 1;
843 Instruction
*mul1
= NULL
; // mul1 before mul2
845 float f
= imm2
.reg
.data
.f32
* exp2f(mul2
->postFactor
);
848 assert(mul2
->op
== OP_MUL
&& mul2
->dType
== TYPE_F32
);
850 if (mul2
->getSrc(t
)->refCount() == 1) {
851 insn
= mul2
->getSrc(t
)->getInsn();
852 if (!mul2
->src(t
).mod
&& insn
->op
== OP_MUL
&& insn
->dType
== TYPE_F32
)
854 if (mul1
&& !mul1
->saturate
) {
857 if (mul1
->src(s1
= 0).getImmediate(imm1
) ||
858 mul1
->src(s1
= 1).getImmediate(imm1
)) {
859 bld
.setPosition(mul1
, false);
861 // d = mul a, imm2 -> d = mul r, (imm1 * imm2)
862 mul1
->setSrc(s1
, bld
.loadImm(NULL
, f
* imm1
.reg
.data
.f32
));
863 mul1
->src(s1
).mod
= Modifier(0);
864 mul2
->def(0).replace(mul1
->getDef(0), false);
865 mul1
->saturate
= mul2
->saturate
;
867 if (prog
->getTarget()->isPostMultiplySupported(OP_MUL
, f
, e
)) {
869 // d = mul c, imm -> d = mul_x_imm a, b
870 mul1
->postFactor
= e
;
871 mul2
->def(0).replace(mul1
->getDef(0), false);
873 mul1
->src(0).mod
*= Modifier(NV50_IR_MOD_NEG
);
874 mul1
->saturate
= mul2
->saturate
;
879 if (mul2
->getDef(0)->refCount() == 1 && !mul2
->saturate
) {
881 // d = mul b, c -> d = mul_x_imm a, c
883 insn
= (*mul2
->getDef(0)->uses
.begin())->getInsn();
888 s2
= insn
->getSrc(0) == mul1
->getDef(0) ? 0 : 1;
890 if (insn
->op
== OP_MUL
&& insn
->dType
== TYPE_F32
)
891 if (!insn
->src(s2
).mod
&& !insn
->src(t2
).getImmediate(imm1
))
893 if (mul2
&& prog
->getTarget()->isPostMultiplySupported(OP_MUL
, f
, e
)) {
894 mul2
->postFactor
= e
;
895 mul2
->setSrc(s2
, mul1
->src(t
));
897 mul2
->src(s2
).mod
*= Modifier(NV50_IR_MOD_NEG
);
903 ConstantFolding::opnd3(Instruction
*i
, ImmediateValue
&imm2
)
908 if (imm2
.isInteger(0)) {
916 if (imm2
.isInteger(0)) {
929 ConstantFolding::opnd(Instruction
*i
, ImmediateValue
&imm0
, int s
)
931 const Target
*target
= prog
->getTarget();
933 const operation op
= i
->op
;
934 Instruction
*newi
= i
;
938 bld
.setPosition(i
, false);
940 uint8_t size
= i
->getDef(0)->reg
.size
;
941 uint8_t bitsize
= size
* 8;
942 uint32_t mask
= (1ULL << bitsize
) - 1;
943 assert(bitsize
<= 32);
945 uint64_t val
= imm0
.reg
.data
.u64
;
946 for (int8_t d
= 0; i
->defExists(d
); ++d
) {
947 Value
*def
= i
->getDef(d
);
948 assert(def
->reg
.size
== size
);
950 newi
= bld
.mkMov(def
, bld
.mkImm((uint32_t)(val
& mask
)), TYPE_U32
);
953 delete_Instruction(prog
, i
);
957 if (i
->dType
== TYPE_F32
)
958 tryCollapseChainedMULs(i
, s
, imm0
);
960 if (i
->subOp
== NV50_IR_SUBOP_MUL_HIGH
) {
961 assert(!isFloatType(i
->sType
));
962 if (imm0
.isInteger(1) && i
->dType
== TYPE_S32
) {
963 bld
.setPosition(i
, false);
964 // Need to set to the sign value, which is a compare.
965 newi
= bld
.mkCmp(OP_SET
, CC_LT
, TYPE_S32
, i
->getDef(0),
966 TYPE_S32
, i
->getSrc(t
), bld
.mkImm(0));
967 delete_Instruction(prog
, i
);
968 } else if (imm0
.isInteger(0) || imm0
.isInteger(1)) {
969 // The high bits can't be set in this case (either mul by 0 or
973 i
->setSrc(0, new_ImmediateValue(prog
, 0u));
974 i
->src(0).mod
= Modifier(0);
976 } else if (!imm0
.isNegative() && imm0
.isPow2()) {
977 // Translate into a shift
981 imm0
.reg
.data
.u32
= 32 - imm0
.reg
.data
.u32
;
982 i
->setSrc(0, i
->getSrc(t
));
983 i
->src(0).mod
= i
->src(t
).mod
;
984 i
->setSrc(1, new_ImmediateValue(prog
, imm0
.reg
.data
.u32
));
988 if (imm0
.isInteger(0)) {
990 i
->setSrc(0, new_ImmediateValue(prog
, 0u));
991 i
->src(0).mod
= Modifier(0);
995 if (!i
->postFactor
&& (imm0
.isInteger(1) || imm0
.isInteger(-1))) {
996 if (imm0
.isNegative())
997 i
->src(t
).mod
= i
->src(t
).mod
^ Modifier(NV50_IR_MOD_NEG
);
998 i
->op
= i
->src(t
).mod
.getOp();
1000 i
->setSrc(0, i
->getSrc(1));
1001 i
->src(0).mod
= i
->src(1).mod
;
1004 if (i
->op
!= OP_CVT
)
1008 if (!i
->postFactor
&& (imm0
.isInteger(2) || imm0
.isInteger(-2))) {
1009 if (imm0
.isNegative())
1010 i
->src(t
).mod
= i
->src(t
).mod
^ Modifier(NV50_IR_MOD_NEG
);
1012 i
->setSrc(s
, i
->getSrc(t
));
1013 i
->src(s
).mod
= i
->src(t
).mod
;
1015 if (!isFloatType(i
->sType
) && !imm0
.isNegative() && imm0
.isPow2()) {
1018 i
->setSrc(0, i
->getSrc(t
));
1019 i
->src(0).mod
= i
->src(t
).mod
;
1020 i
->setSrc(1, new_ImmediateValue(prog
, imm0
.reg
.data
.u32
));
1023 if (i
->postFactor
&& i
->sType
== TYPE_F32
) {
1024 /* Can't emit a postfactor with an immediate, have to fold it in */
1025 i
->setSrc(s
, new_ImmediateValue(
1026 prog
, imm0
.reg
.data
.f32
* exp2f(i
->postFactor
)));
1032 if (imm0
.isInteger(0)) {
1033 i
->setSrc(0, i
->getSrc(2));
1034 i
->src(0).mod
= i
->src(2).mod
;
1037 i
->op
= i
->src(0).mod
.getOp();
1038 if (i
->op
!= OP_CVT
)
1041 if (i
->subOp
!= NV50_IR_SUBOP_MUL_HIGH
&&
1042 (imm0
.isInteger(1) || imm0
.isInteger(-1))) {
1043 if (imm0
.isNegative())
1044 i
->src(t
).mod
= i
->src(t
).mod
^ Modifier(NV50_IR_MOD_NEG
);
1046 i
->setSrc(0, i
->getSrc(1));
1047 i
->src(0).mod
= i
->src(1).mod
;
1049 i
->setSrc(1, i
->getSrc(2));
1050 i
->src(1).mod
= i
->src(2).mod
;
1054 if (s
== 1 && !imm0
.isNegative() && imm0
.isPow2() &&
1055 target
->isOpSupported(OP_SHLADD
, i
->dType
)) {
1058 i
->setSrc(1, new_ImmediateValue(prog
, imm0
.reg
.data
.u32
));
1062 if (imm0
.isInteger(0) && s
== 0 && typeSizeof(i
->dType
) == 8 &&
1063 !isFloatType(i
->dType
))
1069 if (imm0
.isInteger(0)) {
1071 i
->setSrc(0, i
->getSrc(1));
1072 i
->src(0).mod
= i
->src(1).mod
;
1073 if (i
->op
== OP_SUB
)
1074 i
->src(0).mod
= i
->src(0).mod
^ Modifier(NV50_IR_MOD_NEG
);
1077 i
->op
= i
->src(0).mod
.getOp();
1078 if (i
->op
!= OP_CVT
)
1079 i
->src(0).mod
= Modifier(0);
1084 if (s
!= 1 || (i
->dType
!= TYPE_S32
&& i
->dType
!= TYPE_U32
))
1086 bld
.setPosition(i
, false);
1087 if (imm0
.reg
.data
.u32
== 0) {
1090 if (imm0
.reg
.data
.u32
== 1) {
1094 if (i
->dType
== TYPE_U32
&& imm0
.isPow2()) {
1096 i
->setSrc(1, bld
.mkImm(util_logbase2(imm0
.reg
.data
.u32
)));
1098 if (i
->dType
== TYPE_U32
) {
1101 const uint32_t d
= imm0
.reg
.data
.u32
;
1104 uint32_t l
= util_logbase2(d
);
1105 if (((uint32_t)1 << l
) < d
)
1107 m
= (((uint64_t)1 << 32) * (((uint64_t)1 << l
) - d
)) / d
+ 1;
1109 s
= l
? (l
- 1) : 0;
1113 mul
= bld
.mkOp2(OP_MUL
, TYPE_U32
, tA
, i
->getSrc(0),
1114 bld
.loadImm(NULL
, m
));
1115 mul
->subOp
= NV50_IR_SUBOP_MUL_HIGH
;
1116 bld
.mkOp2(OP_SUB
, TYPE_U32
, tB
, i
->getSrc(0), tA
);
1119 bld
.mkOp2(OP_SHR
, TYPE_U32
, tA
, tB
, bld
.mkImm(r
));
1122 tB
= s
? bld
.getSSA() : i
->getDef(0);
1123 newi
= bld
.mkOp2(OP_ADD
, TYPE_U32
, tB
, mul
->getDef(0), tA
);
1125 bld
.mkOp2(OP_SHR
, TYPE_U32
, i
->getDef(0), tB
, bld
.mkImm(s
));
1127 delete_Instruction(prog
, i
);
1129 if (imm0
.reg
.data
.s32
== -1) {
1135 const int32_t d
= imm0
.reg
.data
.s32
;
1137 int32_t l
= util_logbase2(static_cast<unsigned>(abs(d
)));
1138 if ((1 << l
) < abs(d
))
1142 m
= ((uint64_t)1 << (32 + l
- 1)) / abs(d
) + 1 - ((uint64_t)1 << 32);
1146 bld
.mkOp3(OP_MAD
, TYPE_S32
, tA
, i
->getSrc(0), bld
.loadImm(NULL
, m
),
1147 i
->getSrc(0))->subOp
= NV50_IR_SUBOP_MUL_HIGH
;
1149 bld
.mkOp2(OP_SHR
, TYPE_S32
, tB
, tA
, bld
.mkImm(l
- 1));
1153 bld
.mkCmp(OP_SET
, CC_LT
, TYPE_S32
, tA
, TYPE_S32
, i
->getSrc(0), bld
.mkImm(0));
1154 tD
= (d
< 0) ? bld
.getSSA() : i
->getDef(0)->asLValue();
1155 newi
= bld
.mkOp2(OP_SUB
, TYPE_U32
, tD
, tB
, tA
);
1157 bld
.mkOp1(OP_NEG
, TYPE_S32
, i
->getDef(0), tB
);
1159 delete_Instruction(prog
, i
);
1164 if (i
->sType
== TYPE_U32
&& imm0
.isPow2()) {
1165 bld
.setPosition(i
, false);
1167 i
->setSrc(1, bld
.loadImm(NULL
, imm0
.reg
.data
.u32
- 1));
1171 case OP_SET
: // TODO: SET_AND,OR,XOR
1173 /* This optimizes the case where the output of a set is being compared
1174 * to zero. Since the set can only produce 0/-1 (int) or 0/1 (float), we
1175 * can be a lot cleverer in our comparison.
1177 CmpInstruction
*si
= findOriginForTestWithZero(i
->getSrc(t
));
1179 if (imm0
.reg
.data
.u32
!= 0 || !si
)
1182 ccZ
= (CondCode
)((unsigned int)i
->asCmp()->setCond
& ~CC_U
);
1183 // We do everything assuming var (cmp) 0, reverse the condition if 0 is
1186 ccZ
= reverseCondCode(ccZ
);
1187 // If there is a negative modifier, we need to undo that, by flipping
1188 // the comparison to zero.
1189 if (i
->src(t
).mod
.neg())
1190 ccZ
= reverseCondCode(ccZ
);
1191 // If this is a signed comparison, we expect the input to be a regular
1192 // boolean, i.e. 0/-1. However the rest of the logic assumes that true
1193 // is positive, so just flip the sign.
1194 if (i
->sType
== TYPE_S32
) {
1195 assert(!isFloatType(si
->dType
));
1196 ccZ
= reverseCondCode(ccZ
);
1199 case CC_LT
: cc
= CC_FL
; break; // bool < 0 -- this is never true
1200 case CC_GE
: cc
= CC_TR
; break; // bool >= 0 -- this is always true
1201 case CC_EQ
: cc
= inverseCondCode(cc
); break; // bool == 0 -- !bool
1202 case CC_LE
: cc
= inverseCondCode(cc
); break; // bool <= 0 -- !bool
1203 case CC_GT
: break; // bool > 0 -- bool
1204 case CC_NE
: break; // bool != 0 -- bool
1209 // Update the condition of this SET to be identical to the origin set,
1210 // but with the updated condition code. The original SET should get
1213 i
->asCmp()->setCond
= cc
;
1214 i
->setSrc(0, si
->src(0));
1215 i
->setSrc(1, si
->src(1));
1216 if (si
->srcExists(2))
1217 i
->setSrc(2, si
->src(2));
1218 i
->sType
= si
->sType
;
1224 Instruction
*src
= i
->getSrc(t
)->getInsn();
1225 ImmediateValue imm1
;
1226 if (imm0
.reg
.data
.u32
== 0) {
1228 i
->setSrc(0, new_ImmediateValue(prog
, 0u));
1229 i
->src(0).mod
= Modifier(0);
1231 } else if (imm0
.reg
.data
.u32
== ~0U) {
1232 i
->op
= i
->src(t
).mod
.getOp();
1234 i
->setSrc(0, i
->getSrc(t
));
1235 i
->src(0).mod
= i
->src(t
).mod
;
1238 } else if (src
->asCmp()) {
1239 CmpInstruction
*cmp
= src
->asCmp();
1240 if (!cmp
|| cmp
->op
== OP_SLCT
|| cmp
->getDef(0)->refCount() > 1)
1242 if (!prog
->getTarget()->isOpSupported(cmp
->op
, TYPE_F32
))
1244 if (imm0
.reg
.data
.f32
!= 1.0)
1246 if (cmp
->dType
!= TYPE_U32
)
1249 cmp
->dType
= TYPE_F32
;
1250 if (i
->src(t
).mod
!= Modifier(0)) {
1251 assert(i
->src(t
).mod
== Modifier(NV50_IR_MOD_NOT
));
1252 i
->src(t
).mod
= Modifier(0);
1253 cmp
->setCond
= inverseCondCode(cmp
->setCond
);
1258 i
->setSrc(0, i
->getSrc(t
));
1261 } else if (prog
->getTarget()->isOpSupported(OP_EXTBF
, TYPE_U32
) &&
1262 src
->op
== OP_SHR
&&
1263 src
->src(1).getImmediate(imm1
) &&
1264 i
->src(t
).mod
== Modifier(0) &&
1265 util_is_power_of_two(imm0
.reg
.data
.u32
+ 1)) {
1266 // low byte = offset, high byte = width
1267 uint32_t ext
= (util_last_bit(imm0
.reg
.data
.u32
) << 8) | imm1
.reg
.data
.u32
;
1269 i
->setSrc(0, src
->getSrc(0));
1270 i
->setSrc(1, new_ImmediateValue(prog
, ext
));
1271 } else if (src
->op
== OP_SHL
&&
1272 src
->src(1).getImmediate(imm1
) &&
1273 i
->src(t
).mod
== Modifier(0) &&
1274 util_is_power_of_two(~imm0
.reg
.data
.u32
+ 1) &&
1275 util_last_bit(~imm0
.reg
.data
.u32
) <= imm1
.reg
.data
.u32
) {
1279 i
->setSrc(0, i
->getSrc(t
));
1288 if (s
!= 1 || i
->src(0).mod
!= Modifier(0))
1290 // try to concatenate shifts
1291 Instruction
*si
= i
->getSrc(0)->getInsn();
1294 ImmediateValue imm1
;
1297 if (si
->src(1).getImmediate(imm1
)) {
1298 bld
.setPosition(i
, false);
1299 i
->setSrc(0, si
->getSrc(0));
1300 i
->setSrc(1, bld
.loadImm(NULL
, imm0
.reg
.data
.u32
+ imm1
.reg
.data
.u32
));
1304 if (si
->src(1).getImmediate(imm1
) && imm0
.reg
.data
.u32
== imm1
.reg
.data
.u32
) {
1305 bld
.setPosition(i
, false);
1307 i
->setSrc(0, si
->getSrc(0));
1308 i
->setSrc(1, bld
.loadImm(NULL
, ~((1 << imm0
.reg
.data
.u32
) - 1)));
1313 if (isFloatType(si
->dType
))
1315 if (si
->src(1).getImmediate(imm1
))
1317 else if (si
->src(0).getImmediate(imm1
))
1322 bld
.setPosition(i
, false);
1324 i
->setSrc(0, si
->getSrc(!muls
));
1325 i
->setSrc(1, bld
.loadImm(NULL
, imm1
.reg
.data
.u32
<< imm0
.reg
.data
.u32
));
1330 if (isFloatType(si
->dType
))
1332 if (si
->op
!= OP_SUB
&& si
->src(0).getImmediate(imm1
))
1334 else if (si
->src(1).getImmediate(imm1
))
1338 if (si
->src(!adds
).mod
!= Modifier(0))
1340 // SHL(ADD(x, y), z) = ADD(SHL(x, z), SHL(y, z))
1342 // This is more operations, but if one of x, y is an immediate, then
1343 // we can get a situation where (a) we can use ISCADD, or (b)
1344 // propagate the add bit into an indirect load.
1345 bld
.setPosition(i
, false);
1347 i
->setSrc(adds
, bld
.loadImm(NULL
, imm1
.reg
.data
.u32
<< imm0
.reg
.data
.u32
));
1348 i
->setSrc(!adds
, bld
.mkOp2v(OP_SHL
, i
->dType
,
1349 bld
.getSSA(i
->def(0).getSize(), i
->def(0).getFile()),
1351 bld
.mkImm(imm0
.reg
.data
.u32
)));
1376 case TYPE_S32
: res
= util_last_bit_signed(imm0
.reg
.data
.s32
) - 1; break;
1377 case TYPE_U32
: res
= util_last_bit(imm0
.reg
.data
.u32
) - 1; break;
1381 if (i
->subOp
== NV50_IR_SUBOP_BFIND_SAMT
&& res
>= 0)
1383 bld
.setPosition(i
, false); /* make sure bld is init'ed */
1384 i
->setSrc(0, bld
.mkImm(res
));
1391 // Only deal with 1-arg POPCNT here
1392 if (i
->srcExists(1))
1394 uint32_t res
= util_bitcount(imm0
.reg
.data
.u32
);
1395 i
->setSrc(0, new_ImmediateValue(i
->bb
->getProgram(), res
));
1403 // TODO: handle 64-bit values properly
1404 if (typeSizeof(i
->dType
) == 8 || typeSizeof(i
->sType
) == 8)
1407 // TODO: handle single byte/word extractions
1411 bld
.setPosition(i
, true); /* make sure bld is init'ed */
1413 #define CASE(type, dst, fmin, fmax, imin, imax, umin, umax) \
1415 switch (i->sType) { \
1417 res.data.dst = util_iround(i->saturate ? \
1418 CLAMP(imm0.reg.data.f64, fmin, fmax) : \
1419 imm0.reg.data.f64); \
1422 res.data.dst = util_iround(i->saturate ? \
1423 CLAMP(imm0.reg.data.f32, fmin, fmax) : \
1424 imm0.reg.data.f32); \
1427 res.data.dst = i->saturate ? \
1428 CLAMP(imm0.reg.data.s32, imin, imax) : \
1429 imm0.reg.data.s32; \
1432 res.data.dst = i->saturate ? \
1433 CLAMP(imm0.reg.data.u32, umin, umax) : \
1434 imm0.reg.data.u32; \
1437 res.data.dst = i->saturate ? \
1438 CLAMP(imm0.reg.data.s16, imin, imax) : \
1439 imm0.reg.data.s16; \
1442 res.data.dst = i->saturate ? \
1443 CLAMP(imm0.reg.data.u16, umin, umax) : \
1444 imm0.reg.data.u16; \
1448 i->setSrc(0, bld.mkImm(res.data.dst)); \
1452 CASE(TYPE_U16
, u16
, 0, UINT16_MAX
, 0, UINT16_MAX
, 0, UINT16_MAX
);
1453 CASE(TYPE_S16
, s16
, INT16_MIN
, INT16_MAX
, INT16_MIN
, INT16_MAX
, 0, INT16_MAX
);
1454 CASE(TYPE_U32
, u32
, 0, UINT32_MAX
, 0, INT32_MAX
, 0, UINT32_MAX
);
1455 CASE(TYPE_S32
, s32
, INT32_MIN
, INT32_MAX
, INT32_MIN
, INT32_MAX
, 0, INT32_MAX
);
1459 res
.data
.f32
= i
->saturate
?
1460 CLAMP(imm0
.reg
.data
.f64
, 0.0f
, 1.0f
) :
1464 res
.data
.f32
= i
->saturate
?
1465 CLAMP(imm0
.reg
.data
.f32
, 0.0f
, 1.0f
) :
1468 case TYPE_U16
: res
.data
.f32
= (float) imm0
.reg
.data
.u16
; break;
1469 case TYPE_U32
: res
.data
.f32
= (float) imm0
.reg
.data
.u32
; break;
1470 case TYPE_S16
: res
.data
.f32
= (float) imm0
.reg
.data
.s16
; break;
1471 case TYPE_S32
: res
.data
.f32
= (float) imm0
.reg
.data
.s32
; break;
1475 i
->setSrc(0, bld
.mkImm(res
.data
.f32
));
1480 res
.data
.f64
= i
->saturate
?
1481 CLAMP(imm0
.reg
.data
.f64
, 0.0f
, 1.0f
) :
1485 res
.data
.f64
= i
->saturate
?
1486 CLAMP(imm0
.reg
.data
.f32
, 0.0f
, 1.0f
) :
1489 case TYPE_U16
: res
.data
.f64
= (double) imm0
.reg
.data
.u16
; break;
1490 case TYPE_U32
: res
.data
.f64
= (double) imm0
.reg
.data
.u32
; break;
1491 case TYPE_S16
: res
.data
.f64
= (double) imm0
.reg
.data
.s16
; break;
1492 case TYPE_S32
: res
.data
.f64
= (double) imm0
.reg
.data
.s32
; break;
1496 i
->setSrc(0, bld
.mkImm(res
.data
.f64
));
1503 i
->setType(i
->dType
); /* Remove i->sType, which we don't need anymore */
1506 i
->src(0).mod
= Modifier(0); /* Clear the already applied modifier */
1516 // =============================================================================
1518 // Merge modifier operations (ABS, NEG, NOT) into ValueRefs where allowed.
1519 class ModifierFolding
: public Pass
1522 virtual bool visit(BasicBlock
*);
1526 ModifierFolding::visit(BasicBlock
*bb
)
1528 const Target
*target
= prog
->getTarget();
1530 Instruction
*i
, *next
, *mi
;
1533 for (i
= bb
->getEntry(); i
; i
= next
) {
1536 if (0 && i
->op
== OP_SUB
) {
1537 // turn "sub" into "add neg" (do we really want this ?)
1539 i
->src(0).mod
= i
->src(0).mod
^ Modifier(NV50_IR_MOD_NEG
);
1542 for (int s
= 0; s
< 3 && i
->srcExists(s
); ++s
) {
1543 mi
= i
->getSrc(s
)->getInsn();
1545 mi
->predSrc
>= 0 || mi
->getDef(0)->refCount() > 8)
1547 if (i
->sType
== TYPE_U32
&& mi
->dType
== TYPE_S32
) {
1548 if ((i
->op
!= OP_ADD
&&
1550 (mi
->op
!= OP_ABS
&&
1554 if (i
->sType
!= mi
->dType
) {
1557 if ((mod
= Modifier(mi
->op
)) == Modifier(0))
1559 mod
*= mi
->src(0).mod
;
1561 if ((i
->op
== OP_ABS
) || i
->src(s
).mod
.abs()) {
1562 // abs neg [abs] = abs
1563 mod
= mod
& Modifier(~(NV50_IR_MOD_NEG
| NV50_IR_MOD_ABS
));
1565 if ((i
->op
== OP_NEG
) && mod
.neg()) {
1567 // neg as both opcode and modifier on same insn is prohibited
1568 // neg neg abs = abs, neg neg = identity
1569 mod
= mod
& Modifier(~NV50_IR_MOD_NEG
);
1570 i
->op
= mod
.getOp();
1571 mod
= mod
& Modifier(~NV50_IR_MOD_ABS
);
1572 if (mod
== Modifier(0))
1576 if (target
->isModSupported(i
, s
, mod
)) {
1577 i
->setSrc(s
, mi
->getSrc(0));
1578 i
->src(s
).mod
*= mod
;
1582 if (i
->op
== OP_SAT
) {
1583 mi
= i
->getSrc(0)->getInsn();
1585 mi
->getDef(0)->refCount() <= 1 && target
->isSatSupported(mi
)) {
1587 mi
->setDef(0, i
->getDef(0));
1588 delete_Instruction(prog
, i
);
1596 // =============================================================================
1598 // MUL + ADD -> MAD/FMA
1599 // MIN/MAX(a, a) -> a, etc.
1600 // SLCT(a, b, const) -> cc(const) ? a : b
1602 // MUL(MUL(a, b), const) -> MUL_Xconst(a, b)
1603 class AlgebraicOpt
: public Pass
1606 virtual bool visit(BasicBlock
*);
1608 void handleABS(Instruction
*);
1609 bool handleADD(Instruction
*);
1610 bool tryADDToMADOrSAD(Instruction
*, operation toOp
);
1611 void handleMINMAX(Instruction
*);
1612 void handleRCP(Instruction
*);
1613 void handleSLCT(Instruction
*);
1614 void handleLOGOP(Instruction
*);
1615 void handleCVT_NEG(Instruction
*);
1616 void handleCVT_CVT(Instruction
*);
1617 void handleCVT_EXTBF(Instruction
*);
1618 void handleSUCLAMP(Instruction
*);
1619 void handleNEG(Instruction
*);
1625 AlgebraicOpt::handleABS(Instruction
*abs
)
1627 Instruction
*sub
= abs
->getSrc(0)->getInsn();
1630 !prog
->getTarget()->isOpSupported(OP_SAD
, abs
->dType
))
1632 // expect not to have mods yet, if we do, bail
1633 if (sub
->src(0).mod
|| sub
->src(1).mod
)
1635 // hidden conversion ?
1636 ty
= intTypeToSigned(sub
->dType
);
1637 if (abs
->dType
!= abs
->sType
|| ty
!= abs
->sType
)
1640 if ((sub
->op
!= OP_ADD
&& sub
->op
!= OP_SUB
) ||
1641 sub
->src(0).getFile() != FILE_GPR
|| sub
->src(0).mod
||
1642 sub
->src(1).getFile() != FILE_GPR
|| sub
->src(1).mod
)
1645 Value
*src0
= sub
->getSrc(0);
1646 Value
*src1
= sub
->getSrc(1);
1648 if (sub
->op
== OP_ADD
) {
1649 Instruction
*neg
= sub
->getSrc(1)->getInsn();
1650 if (neg
&& neg
->op
!= OP_NEG
) {
1651 neg
= sub
->getSrc(0)->getInsn();
1652 src0
= sub
->getSrc(1);
1654 if (!neg
|| neg
->op
!= OP_NEG
||
1655 neg
->dType
!= neg
->sType
|| neg
->sType
!= ty
)
1657 src1
= neg
->getSrc(0);
1661 abs
->moveSources(1, 2); // move sources >=1 up by 2
1663 abs
->setType(sub
->dType
);
1664 abs
->setSrc(0, src0
);
1665 abs
->setSrc(1, src1
);
1666 bld
.setPosition(abs
, false);
1667 abs
->setSrc(2, bld
.loadImm(bld
.getSSA(typeSizeof(ty
)), 0));
1671 AlgebraicOpt::handleADD(Instruction
*add
)
1673 Value
*src0
= add
->getSrc(0);
1674 Value
*src1
= add
->getSrc(1);
1676 if (src0
->reg
.file
!= FILE_GPR
|| src1
->reg
.file
!= FILE_GPR
)
1679 bool changed
= false;
1680 if (!changed
&& prog
->getTarget()->isOpSupported(OP_MAD
, add
->dType
))
1681 changed
= tryADDToMADOrSAD(add
, OP_MAD
);
1682 if (!changed
&& prog
->getTarget()->isOpSupported(OP_SAD
, add
->dType
))
1683 changed
= tryADDToMADOrSAD(add
, OP_SAD
);
1687 // ADD(SAD(a,b,0), c) -> SAD(a,b,c)
1688 // ADD(MUL(a,b), c) -> MAD(a,b,c)
1690 AlgebraicOpt::tryADDToMADOrSAD(Instruction
*add
, operation toOp
)
1692 Value
*src0
= add
->getSrc(0);
1693 Value
*src1
= add
->getSrc(1);
1696 const operation srcOp
= toOp
== OP_SAD
? OP_SAD
: OP_MUL
;
1697 const Modifier modBad
= Modifier(~((toOp
== OP_MAD
) ? NV50_IR_MOD_NEG
: 0));
1700 if (src0
->refCount() == 1 &&
1701 src0
->getUniqueInsn() && src0
->getUniqueInsn()->op
== srcOp
)
1704 if (src1
->refCount() == 1 &&
1705 src1
->getUniqueInsn() && src1
->getUniqueInsn()->op
== srcOp
)
1710 src
= add
->getSrc(s
);
1712 if (src
->getUniqueInsn() && src
->getUniqueInsn()->bb
!= add
->bb
)
1715 if (src
->getInsn()->saturate
|| src
->getInsn()->postFactor
||
1716 src
->getInsn()->dnz
)
1719 if (toOp
== OP_SAD
) {
1721 if (!src
->getInsn()->src(2).getImmediate(imm
))
1723 if (!imm
.isInteger(0))
1727 if (typeSizeof(add
->dType
) != typeSizeof(src
->getInsn()->dType
) ||
1728 isFloatType(add
->dType
) != isFloatType(src
->getInsn()->dType
))
1731 mod
[0] = add
->src(0).mod
;
1732 mod
[1] = add
->src(1).mod
;
1733 mod
[2] = src
->getUniqueInsn()->src(0).mod
;
1734 mod
[3] = src
->getUniqueInsn()->src(1).mod
;
1736 if (((mod
[0] | mod
[1]) | (mod
[2] | mod
[3])) & modBad
)
1740 add
->subOp
= src
->getInsn()->subOp
; // potentially mul-high
1741 add
->dnz
= src
->getInsn()->dnz
;
1742 add
->dType
= src
->getInsn()->dType
; // sign matters for imad hi
1743 add
->sType
= src
->getInsn()->sType
;
1745 add
->setSrc(2, add
->src(s
? 0 : 1));
1747 add
->setSrc(0, src
->getInsn()->getSrc(0));
1748 add
->src(0).mod
= mod
[2] ^ mod
[s
];
1749 add
->setSrc(1, src
->getInsn()->getSrc(1));
1750 add
->src(1).mod
= mod
[3];
1756 AlgebraicOpt::handleMINMAX(Instruction
*minmax
)
1758 Value
*src0
= minmax
->getSrc(0);
1759 Value
*src1
= minmax
->getSrc(1);
1761 if (src0
!= src1
|| src0
->reg
.file
!= FILE_GPR
)
1763 if (minmax
->src(0).mod
== minmax
->src(1).mod
) {
1764 if (minmax
->def(0).mayReplace(minmax
->src(0))) {
1765 minmax
->def(0).replace(minmax
->src(0), false);
1766 minmax
->bb
->remove(minmax
);
1768 minmax
->op
= OP_CVT
;
1769 minmax
->setSrc(1, NULL
);
1773 // min(x, -x) = -abs(x)
1774 // min(x, -abs(x)) = -abs(x)
1775 // min(x, abs(x)) = x
1776 // max(x, -abs(x)) = x
1777 // max(x, abs(x)) = abs(x)
1778 // max(x, -x) = abs(x)
1783 AlgebraicOpt::handleRCP(Instruction
*rcp
)
1785 Instruction
*si
= rcp
->getSrc(0)->getUniqueInsn();
1787 if (si
&& si
->op
== OP_RCP
) {
1788 Modifier mod
= rcp
->src(0).mod
* si
->src(0).mod
;
1789 rcp
->op
= mod
.getOp();
1790 rcp
->setSrc(0, si
->getSrc(0));
1795 AlgebraicOpt::handleSLCT(Instruction
*slct
)
1797 if (slct
->getSrc(2)->reg
.file
== FILE_IMMEDIATE
) {
1798 if (slct
->getSrc(2)->asImm()->compare(slct
->asCmp()->setCond
, 0.0f
))
1799 slct
->setSrc(0, slct
->getSrc(1));
1801 if (slct
->getSrc(0) != slct
->getSrc(1)) {
1805 slct
->setSrc(1, NULL
);
1806 slct
->setSrc(2, NULL
);
1810 AlgebraicOpt::handleLOGOP(Instruction
*logop
)
1812 Value
*src0
= logop
->getSrc(0);
1813 Value
*src1
= logop
->getSrc(1);
1815 if (src0
->reg
.file
!= FILE_GPR
|| src1
->reg
.file
!= FILE_GPR
)
1819 if ((logop
->op
== OP_AND
|| logop
->op
== OP_OR
) &&
1820 logop
->def(0).mayReplace(logop
->src(0))) {
1821 logop
->def(0).replace(logop
->src(0), false);
1822 delete_Instruction(prog
, logop
);
1825 // try AND(SET, SET) -> SET_AND(SET)
1826 Instruction
*set0
= src0
->getInsn();
1827 Instruction
*set1
= src1
->getInsn();
1829 if (!set0
|| set0
->fixed
|| !set1
|| set1
->fixed
)
1831 if (set1
->op
!= OP_SET
) {
1832 Instruction
*xchg
= set0
;
1835 if (set1
->op
!= OP_SET
)
1838 operation redOp
= (logop
->op
== OP_AND
? OP_SET_AND
:
1839 logop
->op
== OP_XOR
? OP_SET_XOR
: OP_SET_OR
);
1840 if (!prog
->getTarget()->isOpSupported(redOp
, set1
->sType
))
1842 if (set0
->op
!= OP_SET
&&
1843 set0
->op
!= OP_SET_AND
&&
1844 set0
->op
!= OP_SET_OR
&&
1845 set0
->op
!= OP_SET_XOR
)
1847 if (set0
->getDef(0)->refCount() > 1 &&
1848 set1
->getDef(0)->refCount() > 1)
1850 if (set0
->getPredicate() || set1
->getPredicate())
1852 // check that they don't source each other
1853 for (int s
= 0; s
< 2; ++s
)
1854 if (set0
->getSrc(s
) == set1
->getDef(0) ||
1855 set1
->getSrc(s
) == set0
->getDef(0))
1858 set0
= cloneForward(func
, set0
);
1859 set1
= cloneShallow(func
, set1
);
1860 logop
->bb
->insertAfter(logop
, set1
);
1861 logop
->bb
->insertAfter(logop
, set0
);
1863 set0
->dType
= TYPE_U8
;
1864 set0
->getDef(0)->reg
.file
= FILE_PREDICATE
;
1865 set0
->getDef(0)->reg
.size
= 1;
1866 set1
->setSrc(2, set0
->getDef(0));
1868 set1
->setDef(0, logop
->getDef(0));
1869 delete_Instruction(prog
, logop
);
1873 // F2I(NEG(SET with result 1.0f/0.0f)) -> SET with result -1/0
1875 // F2I(NEG(I2F(ABS(SET))))
1877 AlgebraicOpt::handleCVT_NEG(Instruction
*cvt
)
1879 Instruction
*insn
= cvt
->getSrc(0)->getInsn();
1880 if (cvt
->sType
!= TYPE_F32
||
1881 cvt
->dType
!= TYPE_S32
|| cvt
->src(0).mod
!= Modifier(0))
1883 if (!insn
|| insn
->op
!= OP_NEG
|| insn
->dType
!= TYPE_F32
)
1885 if (insn
->src(0).mod
!= Modifier(0))
1887 insn
= insn
->getSrc(0)->getInsn();
1889 // check for nv50 SET(-1,0) -> SET(1.0f/0.0f) chain and nvc0's f32 SET
1890 if (insn
&& insn
->op
== OP_CVT
&&
1891 insn
->dType
== TYPE_F32
&&
1892 insn
->sType
== TYPE_S32
) {
1893 insn
= insn
->getSrc(0)->getInsn();
1894 if (!insn
|| insn
->op
!= OP_ABS
|| insn
->sType
!= TYPE_S32
||
1897 insn
= insn
->getSrc(0)->getInsn();
1898 if (!insn
|| insn
->op
!= OP_SET
|| insn
->dType
!= TYPE_U32
)
1901 if (!insn
|| insn
->op
!= OP_SET
|| insn
->dType
!= TYPE_F32
) {
1905 Instruction
*bset
= cloneShallow(func
, insn
);
1906 bset
->dType
= TYPE_U32
;
1907 bset
->setDef(0, cvt
->getDef(0));
1908 cvt
->bb
->insertAfter(cvt
, bset
);
1909 delete_Instruction(prog
, cvt
);
1912 // F2I(TRUNC()) and so on can be expressed as a single CVT. If the earlier CVT
1913 // does a type conversion, this becomes trickier as there might be range
1914 // changes/etc. We could handle those in theory as long as the range was being
1915 // reduced or kept the same.
1917 AlgebraicOpt::handleCVT_CVT(Instruction
*cvt
)
1919 Instruction
*insn
= cvt
->getSrc(0)->getInsn();
1920 RoundMode rnd
= insn
->rnd
;
1922 if (insn
->saturate
||
1924 insn
->dType
!= insn
->sType
||
1925 insn
->dType
!= cvt
->sType
)
1944 if (!isFloatType(cvt
->dType
) || !isFloatType(insn
->sType
))
1945 rnd
= (RoundMode
)(rnd
& 3);
1948 cvt
->setSrc(0, insn
->getSrc(0));
1949 cvt
->src(0).mod
*= insn
->src(0).mod
;
1950 cvt
->sType
= insn
->sType
;
1953 // Some shaders extract packed bytes out of words and convert them to
1954 // e.g. float. The Fermi+ CVT instruction can extract those directly, as can
1955 // nv50 for word sizes.
1957 // CVT(EXTBF(x, byte/word))
1958 // CVT(AND(bytemask, x))
1959 // CVT(AND(bytemask, SHR(x, 8/16/24)))
1960 // CVT(SHR(x, 16/24))
1962 AlgebraicOpt::handleCVT_EXTBF(Instruction
*cvt
)
1964 Instruction
*insn
= cvt
->getSrc(0)->getInsn();
1967 unsigned width
, offset
;
1968 if ((cvt
->sType
!= TYPE_U32
&& cvt
->sType
!= TYPE_S32
) || !insn
)
1970 if (insn
->op
== OP_EXTBF
&& insn
->src(1).getImmediate(imm
)) {
1971 width
= (imm
.reg
.data
.u32
>> 8) & 0xff;
1972 offset
= imm
.reg
.data
.u32
& 0xff;
1973 arg
= insn
->getSrc(0);
1975 if (width
!= 8 && width
!= 16)
1977 if (width
== 8 && offset
& 0x7)
1979 if (width
== 16 && offset
& 0xf)
1981 } else if (insn
->op
== OP_AND
) {
1983 if (insn
->src(0).getImmediate(imm
))
1985 else if (insn
->src(1).getImmediate(imm
))
1990 if (imm
.reg
.data
.u32
== 0xff)
1992 else if (imm
.reg
.data
.u32
== 0xffff)
1997 arg
= insn
->getSrc(!s
);
1998 Instruction
*shift
= arg
->getInsn();
2000 if (shift
&& shift
->op
== OP_SHR
&&
2001 shift
->sType
== cvt
->sType
&&
2002 shift
->src(1).getImmediate(imm
) &&
2003 ((width
== 8 && (imm
.reg
.data
.u32
& 0x7) == 0) ||
2004 (width
== 16 && (imm
.reg
.data
.u32
& 0xf) == 0))) {
2005 arg
= shift
->getSrc(0);
2006 offset
= imm
.reg
.data
.u32
;
2008 // We just AND'd the high bits away, which means this is effectively an
2010 cvt
->sType
= TYPE_U32
;
2011 } else if (insn
->op
== OP_SHR
&&
2012 insn
->sType
== cvt
->sType
&&
2013 insn
->src(1).getImmediate(imm
)) {
2014 arg
= insn
->getSrc(0);
2015 if (imm
.reg
.data
.u32
== 24) {
2018 } else if (imm
.reg
.data
.u32
== 16) {
2029 // Irrespective of what came earlier, we can undo a shift on the argument
2030 // by adjusting the offset.
2031 Instruction
*shift
= arg
->getInsn();
2032 if (shift
&& shift
->op
== OP_SHL
&&
2033 shift
->src(1).getImmediate(imm
) &&
2034 ((width
== 8 && (imm
.reg
.data
.u32
& 0x7) == 0) ||
2035 (width
== 16 && (imm
.reg
.data
.u32
& 0xf) == 0)) &&
2036 imm
.reg
.data
.u32
<= offset
) {
2037 arg
= shift
->getSrc(0);
2038 offset
-= imm
.reg
.data
.u32
;
2041 // The unpackSnorm lowering still leaves a few shifts behind, but it's too
2042 // annoying to detect them.
2045 cvt
->sType
= cvt
->sType
== TYPE_U32
? TYPE_U8
: TYPE_S8
;
2047 assert(width
== 16);
2048 cvt
->sType
= cvt
->sType
== TYPE_U32
? TYPE_U16
: TYPE_S16
;
2050 cvt
->setSrc(0, arg
);
2051 cvt
->subOp
= offset
>> 3;
2054 // SUCLAMP dst, (ADD b imm), k, 0 -> SUCLAMP dst, b, k, imm (if imm fits s6)
2056 AlgebraicOpt::handleSUCLAMP(Instruction
*insn
)
2059 int32_t val
= insn
->getSrc(2)->asImm()->reg
.data
.s32
;
2063 assert(insn
->srcExists(0) && insn
->src(0).getFile() == FILE_GPR
);
2065 // look for ADD (TODO: only count references by non-SUCLAMP)
2066 if (insn
->getSrc(0)->refCount() > 1)
2068 add
= insn
->getSrc(0)->getInsn();
2069 if (!add
|| add
->op
!= OP_ADD
||
2070 (add
->dType
!= TYPE_U32
&&
2071 add
->dType
!= TYPE_S32
))
2074 // look for immediate
2075 for (s
= 0; s
< 2; ++s
)
2076 if (add
->src(s
).getImmediate(imm
))
2081 // determine if immediate fits
2082 val
+= imm
.reg
.data
.s32
;
2083 if (val
> 31 || val
< -32)
2085 // determine if other addend fits
2086 if (add
->src(s
).getFile() != FILE_GPR
|| add
->src(s
).mod
!= Modifier(0))
2089 bld
.setPosition(insn
, false); // make sure bld is init'ed
2091 insn
->setSrc(2, bld
.mkImm(val
));
2092 insn
->setSrc(0, add
->getSrc(s
));
2095 // NEG(AND(SET, 1)) -> SET
2097 AlgebraicOpt::handleNEG(Instruction
*i
) {
2098 Instruction
*src
= i
->getSrc(0)->getInsn();
2102 if (isFloatType(i
->sType
) || !src
|| src
->op
!= OP_AND
)
2105 if (src
->src(0).getImmediate(imm
))
2107 else if (src
->src(1).getImmediate(imm
))
2112 if (!imm
.isInteger(1))
2115 Instruction
*set
= src
->getSrc(b
)->getInsn();
2116 if ((set
->op
== OP_SET
|| set
->op
== OP_SET_AND
||
2117 set
->op
== OP_SET_OR
|| set
->op
== OP_SET_XOR
) &&
2118 !isFloatType(set
->dType
)) {
2119 i
->def(0).replace(set
->getDef(0), false);
2124 AlgebraicOpt::visit(BasicBlock
*bb
)
2127 for (Instruction
*i
= bb
->getEntry(); i
; i
= next
) {
2154 if (prog
->getTarget()->isOpSupported(OP_EXTBF
, TYPE_U32
))
2171 // =============================================================================
2173 // ADD(SHL(a, b), c) -> SHLADD(a, b, c)
2174 class LateAlgebraicOpt
: public Pass
2177 virtual bool visit(Instruction
*);
2179 void handleADD(Instruction
*);
2180 bool tryADDToSHLADD(Instruction
*);
2184 LateAlgebraicOpt::handleADD(Instruction
*add
)
2186 Value
*src0
= add
->getSrc(0);
2187 Value
*src1
= add
->getSrc(1);
2189 if (src0
->reg
.file
!= FILE_GPR
|| src1
->reg
.file
!= FILE_GPR
)
2192 if (prog
->getTarget()->isOpSupported(OP_SHLADD
, add
->dType
))
2193 tryADDToSHLADD(add
);
2196 // ADD(SHL(a, b), c) -> SHLADD(a, b, c)
2198 LateAlgebraicOpt::tryADDToSHLADD(Instruction
*add
)
2200 Value
*src0
= add
->getSrc(0);
2201 Value
*src1
= add
->getSrc(1);
2207 if (add
->saturate
|| add
->usesFlags() || typeSizeof(add
->dType
) == 8
2208 || isFloatType(add
->dType
))
2211 if (src0
->getUniqueInsn() && src0
->getUniqueInsn()->op
== OP_SHL
)
2214 if (src1
->getUniqueInsn() && src1
->getUniqueInsn()->op
== OP_SHL
)
2219 src
= add
->getSrc(s
);
2220 shl
= src
->getUniqueInsn();
2222 if (shl
->bb
!= add
->bb
|| shl
->usesFlags() || shl
->subOp
|| shl
->src(0).mod
)
2225 if (!shl
->src(1).getImmediate(imm
))
2228 add
->op
= OP_SHLADD
;
2229 add
->setSrc(2, add
->src(!s
));
2230 // SHL can't have any modifiers, but the ADD source may have had
2231 // one. Preserve it.
2232 add
->setSrc(0, shl
->getSrc(0));
2234 add
->src(0).mod
= add
->src(1).mod
;
2235 add
->setSrc(1, new_ImmediateValue(shl
->bb
->getProgram(), imm
.reg
.data
.u32
));
2236 add
->src(1).mod
= Modifier(0);
2242 LateAlgebraicOpt::visit(Instruction
*i
)
2255 // =============================================================================
2257 // Split 64-bit MUL and MAD
2258 class Split64BitOpPreRA
: public Pass
2261 virtual bool visit(BasicBlock
*);
2262 void split64MulMad(Function
*, Instruction
*, DataType
);
2268 Split64BitOpPreRA::visit(BasicBlock
*bb
)
2270 Instruction
*i
, *next
;
2273 for (i
= bb
->getEntry(); i
; i
= next
) {
2278 case TYPE_U64
: hTy
= TYPE_U32
; break;
2279 case TYPE_S64
: hTy
= TYPE_S32
; break;
2284 if (i
->op
== OP_MAD
|| i
->op
== OP_MUL
)
2285 split64MulMad(func
, i
, hTy
);
2292 Split64BitOpPreRA::split64MulMad(Function
*fn
, Instruction
*i
, DataType hTy
)
2294 assert(i
->op
== OP_MAD
|| i
->op
== OP_MUL
);
2295 assert(!isFloatType(i
->dType
) && !isFloatType(i
->sType
));
2296 assert(typeSizeof(hTy
) == 4);
2298 bld
.setPosition(i
, true);
2300 Value
*zero
= bld
.mkImm(0u);
2301 Value
*carry
= bld
.getSSA(1, FILE_FLAGS
);
2303 // We want to compute `d = a * b (+ c)?`, where a, b, c and d are 64-bit
2304 // values (a, b and c might be 32-bit values), using 32-bit operations. This
2305 // gives the following operations:
2306 // * `d.low = low(a.low * b.low) (+ c.low)?`
2307 // * `d.high = low(a.high * b.low) + low(a.low * b.high)
2308 // + high(a.low * b.low) (+ c.high)?`
2310 // To compute the high bits, we can split in the following operations:
2311 // * `tmp1 = low(a.high * b.low) (+ c.high)?`
2312 // * `tmp2 = low(a.low * b.high) + tmp1`
2313 // * `d.high = high(a.low * b.low) + tmp2`
2315 // mkSplit put lower bits at index 0 and higher bits at index 1
2318 if (i
->getSrc(0)->reg
.size
== 8)
2319 bld
.mkSplit(op1
, 4, i
->getSrc(0));
2321 op1
[0] = i
->getSrc(0);
2325 if (i
->getSrc(1)->reg
.size
== 8)
2326 bld
.mkSplit(op2
, 4, i
->getSrc(1));
2328 op2
[0] = i
->getSrc(1);
2332 Value
*op3
[2] = { NULL
, NULL
};
2333 if (i
->op
== OP_MAD
) {
2334 if (i
->getSrc(2)->reg
.size
== 8)
2335 bld
.mkSplit(op3
, 4, i
->getSrc(2));
2337 op3
[0] = i
->getSrc(2);
2342 Value
*tmpRes1Hi
= bld
.getSSA();
2343 if (i
->op
== OP_MAD
)
2344 bld
.mkOp3(OP_MAD
, hTy
, tmpRes1Hi
, op1
[1], op2
[0], op3
[1]);
2346 bld
.mkOp2(OP_MUL
, hTy
, tmpRes1Hi
, op1
[1], op2
[0]);
2348 Value
*tmpRes2Hi
= bld
.mkOp3v(OP_MAD
, hTy
, bld
.getSSA(), op1
[0], op2
[1], tmpRes1Hi
);
2350 Value
*def
[2] = { bld
.getSSA(), bld
.getSSA() };
2352 // If it was a MAD, add the carry from the low bits
2353 // It is not needed if it was a MUL, since we added high(a.low * b.low) to
2355 if (i
->op
== OP_MAD
)
2356 bld
.mkOp3(OP_MAD
, hTy
, def
[0], op1
[0], op2
[0], op3
[0])->setFlagsDef(1, carry
);
2358 bld
.mkOp2(OP_MUL
, hTy
, def
[0], op1
[0], op2
[0]);
2360 Instruction
*hiPart3
= bld
.mkOp3(OP_MAD
, hTy
, def
[1], op1
[0], op2
[0], tmpRes2Hi
);
2361 hiPart3
->subOp
= NV50_IR_SUBOP_MUL_HIGH
;
2362 if (i
->op
== OP_MAD
)
2363 hiPart3
->setFlagsSrc(3, carry
);
2365 bld
.mkOp2(OP_MERGE
, i
->dType
, i
->getDef(0), def
[0], def
[1]);
2367 delete_Instruction(fn
->getProgram(), i
);
2370 // =============================================================================
2373 updateLdStOffset(Instruction
*ldst
, int32_t offset
, Function
*fn
)
2375 if (offset
!= ldst
->getSrc(0)->reg
.data
.offset
) {
2376 if (ldst
->getSrc(0)->refCount() > 1)
2377 ldst
->setSrc(0, cloneShallow(fn
, ldst
->getSrc(0)));
2378 ldst
->getSrc(0)->reg
.data
.offset
= offset
;
2382 // Combine loads and stores, forward stores to loads where possible.
2383 class MemoryOpt
: public Pass
2391 const Value
*rel
[2];
2399 bool overlaps(const Instruction
*ldst
) const;
2401 inline void link(Record
**);
2402 inline void unlink(Record
**);
2403 inline void set(const Instruction
*ldst
);
2409 Record
*loads
[DATA_FILE_COUNT
];
2410 Record
*stores
[DATA_FILE_COUNT
];
2412 MemoryPool recordPool
;
2415 virtual bool visit(BasicBlock
*);
2416 bool runOpt(BasicBlock
*);
2418 Record
**getList(const Instruction
*);
2420 Record
*findRecord(const Instruction
*, bool load
, bool& isAdjacent
) const;
2422 // merge @insn into load/store instruction from @rec
2423 bool combineLd(Record
*rec
, Instruction
*ld
);
2424 bool combineSt(Record
*rec
, Instruction
*st
);
2426 bool replaceLdFromLd(Instruction
*ld
, Record
*ldRec
);
2427 bool replaceLdFromSt(Instruction
*ld
, Record
*stRec
);
2428 bool replaceStFromSt(Instruction
*restrict st
, Record
*stRec
);
2430 void addRecord(Instruction
*ldst
);
2431 void purgeRecords(Instruction
*const st
, DataFile
);
2432 void lockStores(Instruction
*const ld
);
2439 MemoryOpt::MemoryOpt() : recordPool(sizeof(MemoryOpt::Record
), 6)
2441 for (int i
= 0; i
< DATA_FILE_COUNT
; ++i
) {
2451 for (unsigned int i
= 0; i
< DATA_FILE_COUNT
; ++i
) {
2453 for (it
= loads
[i
]; it
; it
= next
) {
2455 recordPool
.release(it
);
2458 for (it
= stores
[i
]; it
; it
= next
) {
2460 recordPool
.release(it
);
2467 MemoryOpt::combineLd(Record
*rec
, Instruction
*ld
)
2469 int32_t offRc
= rec
->offset
;
2470 int32_t offLd
= ld
->getSrc(0)->reg
.data
.offset
;
2471 int sizeRc
= rec
->size
;
2472 int sizeLd
= typeSizeof(ld
->dType
);
2473 int size
= sizeRc
+ sizeLd
;
2476 if (!prog
->getTarget()->
2477 isAccessSupported(ld
->getSrc(0)->reg
.file
, typeOfSize(size
)))
2479 // no unaligned loads
2480 if (((size
== 0x8) && (MIN2(offLd
, offRc
) & 0x7)) ||
2481 ((size
== 0xc) && (MIN2(offLd
, offRc
) & 0xf)))
2483 // for compute indirect loads are not guaranteed to be aligned
2484 if (prog
->getType() == Program::TYPE_COMPUTE
&& rec
->rel
[0])
2487 assert(sizeRc
+ sizeLd
<= 16 && offRc
!= offLd
);
2489 for (j
= 0; sizeRc
; sizeRc
-= rec
->insn
->getDef(j
)->reg
.size
, ++j
);
2491 if (offLd
< offRc
) {
2493 for (sz
= 0, d
= 0; sz
< sizeLd
; sz
+= ld
->getDef(d
)->reg
.size
, ++d
);
2494 // d: nr of definitions in ld
2495 // j: nr of definitions in rec->insn, move:
2496 for (d
= d
+ j
- 1; j
> 0; --j
, --d
)
2497 rec
->insn
->setDef(d
, rec
->insn
->getDef(j
- 1));
2499 if (rec
->insn
->getSrc(0)->refCount() > 1)
2500 rec
->insn
->setSrc(0, cloneShallow(func
, rec
->insn
->getSrc(0)));
2501 rec
->offset
= rec
->insn
->getSrc(0)->reg
.data
.offset
= offLd
;
2507 // move definitions of @ld to @rec->insn
2508 for (j
= 0; sizeLd
; ++j
, ++d
) {
2509 sizeLd
-= ld
->getDef(j
)->reg
.size
;
2510 rec
->insn
->setDef(d
, ld
->getDef(j
));
2514 rec
->insn
->getSrc(0)->reg
.size
= size
;
2515 rec
->insn
->setType(typeOfSize(size
));
2517 delete_Instruction(prog
, ld
);
2523 MemoryOpt::combineSt(Record
*rec
, Instruction
*st
)
2525 int32_t offRc
= rec
->offset
;
2526 int32_t offSt
= st
->getSrc(0)->reg
.data
.offset
;
2527 int sizeRc
= rec
->size
;
2528 int sizeSt
= typeSizeof(st
->dType
);
2530 int size
= sizeRc
+ sizeSt
;
2532 Value
*src
[4]; // no modifiers in ValueRef allowed for st
2535 if (!prog
->getTarget()->
2536 isAccessSupported(st
->getSrc(0)->reg
.file
, typeOfSize(size
)))
2538 // no unaligned stores
2539 if (size
== 8 && MIN2(offRc
, offSt
) & 0x7)
2541 // for compute indirect stores are not guaranteed to be aligned
2542 if (prog
->getType() == Program::TYPE_COMPUTE
&& rec
->rel
[0])
2545 st
->takeExtraSources(0, extra
); // save predicate and indirect address
2547 if (offRc
< offSt
) {
2548 // save values from @st
2549 for (s
= 0; sizeSt
; ++s
) {
2550 sizeSt
-= st
->getSrc(s
+ 1)->reg
.size
;
2551 src
[s
] = st
->getSrc(s
+ 1);
2553 // set record's values as low sources of @st
2554 for (j
= 1; sizeRc
; ++j
) {
2555 sizeRc
-= rec
->insn
->getSrc(j
)->reg
.size
;
2556 st
->setSrc(j
, rec
->insn
->getSrc(j
));
2558 // set saved values as high sources of @st
2559 for (k
= j
, j
= 0; j
< s
; ++j
)
2560 st
->setSrc(k
++, src
[j
]);
2562 updateLdStOffset(st
, offRc
, func
);
2564 for (j
= 1; sizeSt
; ++j
)
2565 sizeSt
-= st
->getSrc(j
)->reg
.size
;
2566 for (s
= 1; sizeRc
; ++j
, ++s
) {
2567 sizeRc
-= rec
->insn
->getSrc(s
)->reg
.size
;
2568 st
->setSrc(j
, rec
->insn
->getSrc(s
));
2570 rec
->offset
= offSt
;
2572 st
->putExtraSources(0, extra
); // restore pointer and predicate
2574 delete_Instruction(prog
, rec
->insn
);
2577 rec
->insn
->getSrc(0)->reg
.size
= size
;
2578 rec
->insn
->setType(typeOfSize(size
));
2583 MemoryOpt::Record::set(const Instruction
*ldst
)
2585 const Symbol
*mem
= ldst
->getSrc(0)->asSym();
2586 fileIndex
= mem
->reg
.fileIndex
;
2587 rel
[0] = ldst
->getIndirect(0, 0);
2588 rel
[1] = ldst
->getIndirect(0, 1);
2589 offset
= mem
->reg
.data
.offset
;
2590 base
= mem
->getBase();
2591 size
= typeSizeof(ldst
->sType
);
2595 MemoryOpt::Record::link(Record
**list
)
2605 MemoryOpt::Record::unlink(Record
**list
)
2615 MemoryOpt::Record
**
2616 MemoryOpt::getList(const Instruction
*insn
)
2618 if (insn
->op
== OP_LOAD
|| insn
->op
== OP_VFETCH
)
2619 return &loads
[insn
->src(0).getFile()];
2620 return &stores
[insn
->src(0).getFile()];
2624 MemoryOpt::addRecord(Instruction
*i
)
2626 Record
**list
= getList(i
);
2627 Record
*it
= reinterpret_cast<Record
*>(recordPool
.allocate());
2636 MemoryOpt::findRecord(const Instruction
*insn
, bool load
, bool& isAdj
) const
2638 const Symbol
*sym
= insn
->getSrc(0)->asSym();
2639 const int size
= typeSizeof(insn
->sType
);
2641 Record
*it
= load
? loads
[sym
->reg
.file
] : stores
[sym
->reg
.file
];
2643 for (; it
; it
= it
->next
) {
2644 if (it
->locked
&& insn
->op
!= OP_LOAD
&& insn
->op
!= OP_VFETCH
)
2646 if ((it
->offset
>> 4) != (sym
->reg
.data
.offset
>> 4) ||
2647 it
->rel
[0] != insn
->getIndirect(0, 0) ||
2648 it
->fileIndex
!= sym
->reg
.fileIndex
||
2649 it
->rel
[1] != insn
->getIndirect(0, 1))
2652 if (it
->offset
< sym
->reg
.data
.offset
) {
2653 if (it
->offset
+ it
->size
>= sym
->reg
.data
.offset
) {
2654 isAdj
= (it
->offset
+ it
->size
== sym
->reg
.data
.offset
);
2657 if (!(it
->offset
& 0x7))
2661 isAdj
= it
->offset
!= sym
->reg
.data
.offset
;
2662 if (size
<= it
->size
&& !isAdj
)
2665 if (!(sym
->reg
.data
.offset
& 0x7))
2666 if (it
->offset
- size
<= sym
->reg
.data
.offset
)
2674 MemoryOpt::replaceLdFromSt(Instruction
*ld
, Record
*rec
)
2676 Instruction
*st
= rec
->insn
;
2677 int32_t offSt
= rec
->offset
;
2678 int32_t offLd
= ld
->getSrc(0)->reg
.data
.offset
;
2681 for (s
= 1; offSt
!= offLd
&& st
->srcExists(s
); ++s
)
2682 offSt
+= st
->getSrc(s
)->reg
.size
;
2686 for (d
= 0; ld
->defExists(d
) && st
->srcExists(s
); ++d
, ++s
) {
2687 if (ld
->getDef(d
)->reg
.size
!= st
->getSrc(s
)->reg
.size
)
2689 if (st
->getSrc(s
)->reg
.file
!= FILE_GPR
)
2691 ld
->def(d
).replace(st
->src(s
), false);
2698 MemoryOpt::replaceLdFromLd(Instruction
*ldE
, Record
*rec
)
2700 Instruction
*ldR
= rec
->insn
;
2701 int32_t offR
= rec
->offset
;
2702 int32_t offE
= ldE
->getSrc(0)->reg
.data
.offset
;
2705 assert(offR
<= offE
);
2706 for (dR
= 0; offR
< offE
&& ldR
->defExists(dR
); ++dR
)
2707 offR
+= ldR
->getDef(dR
)->reg
.size
;
2711 for (dE
= 0; ldE
->defExists(dE
) && ldR
->defExists(dR
); ++dE
, ++dR
) {
2712 if (ldE
->getDef(dE
)->reg
.size
!= ldR
->getDef(dR
)->reg
.size
)
2714 ldE
->def(dE
).replace(ldR
->getDef(dR
), false);
2717 delete_Instruction(prog
, ldE
);
2722 MemoryOpt::replaceStFromSt(Instruction
*restrict st
, Record
*rec
)
2724 const Instruction
*const ri
= rec
->insn
;
2727 int32_t offS
= st
->getSrc(0)->reg
.data
.offset
;
2728 int32_t offR
= rec
->offset
;
2729 int32_t endS
= offS
+ typeSizeof(st
->dType
);
2730 int32_t endR
= offR
+ typeSizeof(ri
->dType
);
2732 rec
->size
= MAX2(endS
, endR
) - MIN2(offS
, offR
);
2734 st
->takeExtraSources(0, extra
);
2740 // get non-replaced sources of ri
2741 for (s
= 1; offR
< offS
; offR
+= ri
->getSrc(s
)->reg
.size
, ++s
)
2742 vals
[k
++] = ri
->getSrc(s
);
2744 // get replaced sources of st
2745 for (s
= 1; st
->srcExists(s
); offS
+= st
->getSrc(s
)->reg
.size
, ++s
)
2746 vals
[k
++] = st
->getSrc(s
);
2747 // skip replaced sources of ri
2748 for (s
= n
; offR
< endS
; offR
+= ri
->getSrc(s
)->reg
.size
, ++s
);
2749 // get non-replaced sources after values covered by st
2750 for (; offR
< endR
; offR
+= ri
->getSrc(s
)->reg
.size
, ++s
)
2751 vals
[k
++] = ri
->getSrc(s
);
2752 assert((unsigned int)k
<= ARRAY_SIZE(vals
));
2753 for (s
= 0; s
< k
; ++s
)
2754 st
->setSrc(s
+ 1, vals
[s
]);
2755 st
->setSrc(0, ri
->getSrc(0));
2759 for (j
= 1; offR
< endS
; offR
+= ri
->getSrc(j
++)->reg
.size
);
2760 for (s
= 1; offS
< endS
; offS
+= st
->getSrc(s
++)->reg
.size
);
2761 for (; offR
< endR
; offR
+= ri
->getSrc(j
++)->reg
.size
)
2762 st
->setSrc(s
++, ri
->getSrc(j
));
2764 st
->putExtraSources(0, extra
);
2766 delete_Instruction(prog
, rec
->insn
);
2769 rec
->offset
= st
->getSrc(0)->reg
.data
.offset
;
2771 st
->setType(typeOfSize(rec
->size
));
2777 MemoryOpt::Record::overlaps(const Instruction
*ldst
) const
2782 // This assumes that images/buffers can't overlap. They can.
2783 // TODO: Plumb the restrict logic through, and only skip when it's a
2784 // restrict situation, or there can implicitly be no writes.
2785 if (this->fileIndex
!= that
.fileIndex
&& this->rel
[1] == that
.rel
[1])
2788 if (this->rel
[0] || that
.rel
[0])
2789 return this->base
== that
.base
;
2792 (this->offset
< that
.offset
+ that
.size
) &&
2793 (this->offset
+ this->size
> that
.offset
);
2796 // We must not eliminate stores that affect the result of @ld if
2797 // we find later stores to the same location, and we may no longer
2798 // merge them with later stores.
2799 // The stored value can, however, still be used to determine the value
2800 // returned by future loads.
2802 MemoryOpt::lockStores(Instruction
*const ld
)
2804 for (Record
*r
= stores
[ld
->src(0).getFile()]; r
; r
= r
->next
)
2805 if (!r
->locked
&& r
->overlaps(ld
))
2809 // Prior loads from the location of @st are no longer valid.
2810 // Stores to the location of @st may no longer be used to derive
2811 // the value at it nor be coalesced into later stores.
2813 MemoryOpt::purgeRecords(Instruction
*const st
, DataFile f
)
2816 f
= st
->src(0).getFile();
2818 for (Record
*r
= loads
[f
]; r
; r
= r
->next
)
2819 if (!st
|| r
->overlaps(st
))
2820 r
->unlink(&loads
[f
]);
2822 for (Record
*r
= stores
[f
]; r
; r
= r
->next
)
2823 if (!st
|| r
->overlaps(st
))
2824 r
->unlink(&stores
[f
]);
2828 MemoryOpt::visit(BasicBlock
*bb
)
2830 bool ret
= runOpt(bb
);
2831 // Run again, one pass won't combine 4 32 bit ld/st to a single 128 bit ld/st
2832 // where 96 bit memory operations are forbidden.
2839 MemoryOpt::runOpt(BasicBlock
*bb
)
2841 Instruction
*ldst
, *next
;
2843 bool isAdjacent
= true;
2845 for (ldst
= bb
->getEntry(); ldst
; ldst
= next
) {
2850 if (ldst
->op
== OP_LOAD
|| ldst
->op
== OP_VFETCH
) {
2851 if (ldst
->isDead()) {
2852 // might have been produced by earlier optimization
2853 delete_Instruction(prog
, ldst
);
2857 if (ldst
->op
== OP_STORE
|| ldst
->op
== OP_EXPORT
) {
2858 if (typeSizeof(ldst
->dType
) == 4 &&
2859 ldst
->src(1).getFile() == FILE_GPR
&&
2860 ldst
->getSrc(1)->getInsn()->op
== OP_NOP
) {
2861 delete_Instruction(prog
, ldst
);
2866 // TODO: maybe have all fixed ops act as barrier ?
2867 if (ldst
->op
== OP_CALL
||
2868 ldst
->op
== OP_BAR
||
2869 ldst
->op
== OP_MEMBAR
) {
2870 purgeRecords(NULL
, FILE_MEMORY_LOCAL
);
2871 purgeRecords(NULL
, FILE_MEMORY_GLOBAL
);
2872 purgeRecords(NULL
, FILE_MEMORY_SHARED
);
2873 purgeRecords(NULL
, FILE_SHADER_OUTPUT
);
2875 if (ldst
->op
== OP_ATOM
|| ldst
->op
== OP_CCTL
) {
2876 if (ldst
->src(0).getFile() == FILE_MEMORY_GLOBAL
) {
2877 purgeRecords(NULL
, FILE_MEMORY_LOCAL
);
2878 purgeRecords(NULL
, FILE_MEMORY_GLOBAL
);
2879 purgeRecords(NULL
, FILE_MEMORY_SHARED
);
2881 purgeRecords(NULL
, ldst
->src(0).getFile());
2884 if (ldst
->op
== OP_EMIT
|| ldst
->op
== OP_RESTART
) {
2885 purgeRecords(NULL
, FILE_SHADER_OUTPUT
);
2889 if (ldst
->getPredicate()) // TODO: handle predicated ld/st
2891 if (ldst
->perPatch
) // TODO: create separate per-patch lists
2895 DataFile file
= ldst
->src(0).getFile();
2897 // if ld l[]/g[] look for previous store to eliminate the reload
2898 if (file
== FILE_MEMORY_GLOBAL
|| file
== FILE_MEMORY_LOCAL
) {
2899 // TODO: shared memory ?
2900 rec
= findRecord(ldst
, false, isAdjacent
);
2901 if (rec
&& !isAdjacent
)
2902 keep
= !replaceLdFromSt(ldst
, rec
);
2905 // or look for ld from the same location and replace this one
2906 rec
= keep
? findRecord(ldst
, true, isAdjacent
) : NULL
;
2909 keep
= !replaceLdFromLd(ldst
, rec
);
2911 // or combine a previous load with this one
2912 keep
= !combineLd(rec
, ldst
);
2917 rec
= findRecord(ldst
, false, isAdjacent
);
2920 keep
= !replaceStFromSt(ldst
, rec
);
2922 keep
= !combineSt(rec
, ldst
);
2925 purgeRecords(ldst
, DATA_FILE_COUNT
);
2935 // =============================================================================
2937 // Turn control flow into predicated instructions (after register allocation !).
2939 // Could move this to before register allocation on NVC0 and also handle nested
2941 class FlatteningPass
: public Pass
2944 virtual bool visit(Function
*);
2945 virtual bool visit(BasicBlock
*);
2947 bool tryPredicateConditional(BasicBlock
*);
2948 void predicateInstructions(BasicBlock
*, Value
*pred
, CondCode cc
);
2949 void tryPropagateBranch(BasicBlock
*);
2950 inline bool isConstantCondition(Value
*pred
);
2951 inline bool mayPredicate(const Instruction
*, const Value
*pred
) const;
2952 inline void removeFlow(Instruction
*);
2958 FlatteningPass::isConstantCondition(Value
*pred
)
2960 Instruction
*insn
= pred
->getUniqueInsn();
2962 if (insn
->op
!= OP_SET
|| insn
->srcExists(2))
2965 for (int s
= 0; s
< 2 && insn
->srcExists(s
); ++s
) {
2966 Instruction
*ld
= insn
->getSrc(s
)->getUniqueInsn();
2969 if (ld
->op
!= OP_MOV
&& ld
->op
!= OP_LOAD
)
2971 if (ld
->src(0).isIndirect(0))
2973 file
= ld
->src(0).getFile();
2975 file
= insn
->src(s
).getFile();
2976 // catch $r63 on NVC0 and $r63/$r127 on NV50. Unfortunately maxGPR is
2977 // in register "units", which can vary between targets.
2978 if (file
== FILE_GPR
) {
2979 Value
*v
= insn
->getSrc(s
);
2980 int bytes
= v
->reg
.data
.id
* MIN2(v
->reg
.size
, 4);
2981 int units
= bytes
>> gpr_unit
;
2982 if (units
> prog
->maxGPR
)
2983 file
= FILE_IMMEDIATE
;
2986 if (file
!= FILE_IMMEDIATE
&& file
!= FILE_MEMORY_CONST
)
2993 FlatteningPass::removeFlow(Instruction
*insn
)
2995 FlowInstruction
*term
= insn
? insn
->asFlow() : NULL
;
2998 Graph::Edge::Type ty
= term
->bb
->cfg
.outgoing().getType();
3000 if (term
->op
== OP_BRA
) {
3001 // TODO: this might get more difficult when we get arbitrary BRAs
3002 if (ty
== Graph::Edge::CROSS
|| ty
== Graph::Edge::BACK
)
3005 if (term
->op
!= OP_JOIN
)
3008 Value
*pred
= term
->getPredicate();
3010 delete_Instruction(prog
, term
);
3012 if (pred
&& pred
->refCount() == 0) {
3013 Instruction
*pSet
= pred
->getUniqueInsn();
3014 pred
->join
->reg
.data
.id
= -1; // deallocate
3016 delete_Instruction(prog
, pSet
);
3021 FlatteningPass::predicateInstructions(BasicBlock
*bb
, Value
*pred
, CondCode cc
)
3023 for (Instruction
*i
= bb
->getEntry(); i
; i
= i
->next
) {
3026 assert(!i
->getPredicate());
3027 i
->setPredicate(cc
, pred
);
3029 removeFlow(bb
->getExit());
3033 FlatteningPass::mayPredicate(const Instruction
*insn
, const Value
*pred
) const
3035 if (insn
->isPseudo())
3037 // TODO: calls where we don't know which registers are modified
3039 if (!prog
->getTarget()->mayPredicate(insn
, pred
))
3041 for (int d
= 0; insn
->defExists(d
); ++d
)
3042 if (insn
->getDef(d
)->equals(pred
))
3047 // If we jump to BRA/RET/EXIT, replace the jump with it.
3048 // NOTE: We do not update the CFG anymore here !
3050 // TODO: Handle cases where we skip over a branch (maybe do that elsewhere ?):
3052 // @p0 bra BB:2 -> @!p0 bra BB:3 iff (!) BB:2 immediately adjoins BB:1
3060 FlatteningPass::tryPropagateBranch(BasicBlock
*bb
)
3062 for (Instruction
*i
= bb
->getExit(); i
&& i
->op
== OP_BRA
; i
= i
->prev
) {
3063 BasicBlock
*bf
= i
->asFlow()->target
.bb
;
3065 if (bf
->getInsnCount() != 1)
3068 FlowInstruction
*bra
= i
->asFlow();
3069 FlowInstruction
*rep
= bf
->getExit()->asFlow();
3071 if (!rep
|| rep
->getPredicate())
3073 if (rep
->op
!= OP_BRA
&&
3074 rep
->op
!= OP_JOIN
&&
3078 // TODO: If there are multiple branches to @rep, only the first would
3079 // be replaced, so only remove them after this pass is done ?
3080 // Also, need to check all incident blocks for fall-through exits and
3081 // add the branch there.
3083 bra
->target
.bb
= rep
->target
.bb
;
3084 if (bf
->cfg
.incidentCount() == 1)
3090 FlatteningPass::visit(Function
*fn
)
3092 gpr_unit
= prog
->getTarget()->getFileUnit(FILE_GPR
);
3098 FlatteningPass::visit(BasicBlock
*bb
)
3100 if (tryPredicateConditional(bb
))
3103 // try to attach join to previous instruction
3104 if (prog
->getTarget()->hasJoin
) {
3105 Instruction
*insn
= bb
->getExit();
3106 if (insn
&& insn
->op
== OP_JOIN
&& !insn
->getPredicate()) {
3108 if (insn
&& !insn
->getPredicate() &&
3110 insn
->op
!= OP_DISCARD
&&
3111 insn
->op
!= OP_TEXBAR
&&
3112 !isTextureOp(insn
->op
) && // probably just nve4
3113 !isSurfaceOp(insn
->op
) && // not confirmed
3114 insn
->op
!= OP_LINTERP
&& // probably just nve4
3115 insn
->op
!= OP_PINTERP
&& // probably just nve4
3116 ((insn
->op
!= OP_LOAD
&& insn
->op
!= OP_STORE
&& insn
->op
!= OP_ATOM
) ||
3117 (typeSizeof(insn
->dType
) <= 4 && !insn
->src(0).isIndirect(0))) &&
3120 bb
->remove(bb
->getExit());
3126 tryPropagateBranch(bb
);
3132 FlatteningPass::tryPredicateConditional(BasicBlock
*bb
)
3134 BasicBlock
*bL
= NULL
, *bR
= NULL
;
3135 unsigned int nL
= 0, nR
= 0, limit
= 12;
3139 mask
= bb
->initiatesSimpleConditional();
3143 assert(bb
->getExit());
3144 Value
*pred
= bb
->getExit()->getPredicate();
3147 if (isConstantCondition(pred
))
3150 Graph::EdgeIterator ei
= bb
->cfg
.outgoing();
3153 bL
= BasicBlock::get(ei
.getNode());
3154 for (insn
= bL
->getEntry(); insn
; insn
= insn
->next
, ++nL
)
3155 if (!mayPredicate(insn
, pred
))
3158 return false; // too long, do a real branch
3163 bR
= BasicBlock::get(ei
.getNode());
3164 for (insn
= bR
->getEntry(); insn
; insn
= insn
->next
, ++nR
)
3165 if (!mayPredicate(insn
, pred
))
3168 return false; // too long, do a real branch
3172 predicateInstructions(bL
, pred
, bb
->getExit()->cc
);
3174 predicateInstructions(bR
, pred
, inverseCondCode(bb
->getExit()->cc
));
3177 bb
->remove(bb
->joinAt
);
3180 removeFlow(bb
->getExit()); // delete the branch/join at the fork point
3182 // remove potential join operations at the end of the conditional
3183 if (prog
->getTarget()->joinAnterior
) {
3184 bb
= BasicBlock::get((bL
? bL
: bR
)->cfg
.outgoing().getNode());
3185 if (bb
->getEntry() && bb
->getEntry()->op
== OP_JOIN
)
3186 removeFlow(bb
->getEntry());
3192 // =============================================================================
3194 // Fold Immediate into MAD; must be done after register allocation due to
3195 // constraint SDST == SSRC2
3197 // Does NVC0+ have other situations where this pass makes sense?
3198 class PostRaLoadPropagation
: public Pass
3201 virtual bool visit(Instruction
*);
3203 void handleMADforNV50(Instruction
*);
3204 void handleMADforNVC0(Instruction
*);
3208 post_ra_dead(Instruction
*i
)
3210 for (int d
= 0; i
->defExists(d
); ++d
)
3211 if (i
->getDef(d
)->refCount())
3216 // Fold Immediate into MAD; must be done after register allocation due to
3217 // constraint SDST == SSRC2
3219 PostRaLoadPropagation::handleMADforNV50(Instruction
*i
)
3221 if (i
->def(0).getFile() != FILE_GPR
||
3222 i
->src(0).getFile() != FILE_GPR
||
3223 i
->src(1).getFile() != FILE_GPR
||
3224 i
->src(2).getFile() != FILE_GPR
||
3225 i
->getDef(0)->reg
.data
.id
!= i
->getSrc(2)->reg
.data
.id
)
3228 if (i
->getDef(0)->reg
.data
.id
>= 64 ||
3229 i
->getSrc(0)->reg
.data
.id
>= 64)
3232 if (i
->flagsSrc
>= 0 && i
->getSrc(i
->flagsSrc
)->reg
.data
.id
!= 0)
3235 if (i
->getPredicate())
3239 Instruction
*def
= i
->getSrc(1)->getInsn();
3241 if (def
&& def
->op
== OP_SPLIT
&& typeSizeof(def
->sType
) == 4)
3242 def
= def
->getSrc(0)->getInsn();
3243 if (def
&& def
->op
== OP_MOV
&& def
->src(0).getFile() == FILE_IMMEDIATE
) {
3244 vtmp
= i
->getSrc(1);
3245 if (isFloatType(i
->sType
)) {
3246 i
->setSrc(1, def
->getSrc(0));
3249 bool ret
= def
->src(0).getImmediate(val
);
3251 if (i
->getSrc(1)->reg
.data
.id
& 1)
3252 val
.reg
.data
.u32
>>= 16;
3253 val
.reg
.data
.u32
&= 0xffff;
3254 i
->setSrc(1, new_ImmediateValue(prog
, val
.reg
.data
.u32
));
3257 /* There's no post-RA dead code elimination, so do it here
3258 * XXX: if we add more code-removing post-RA passes, we might
3259 * want to create a post-RA dead-code elim pass */
3260 if (post_ra_dead(vtmp
->getInsn())) {
3261 Value
*src
= vtmp
->getInsn()->getSrc(0);
3262 // Careful -- splits will have already been removed from the
3263 // functions. Don't double-delete.
3264 if (vtmp
->getInsn()->bb
)
3265 delete_Instruction(prog
, vtmp
->getInsn());
3266 if (src
->getInsn() && post_ra_dead(src
->getInsn()))
3267 delete_Instruction(prog
, src
->getInsn());
3273 PostRaLoadPropagation::handleMADforNVC0(Instruction
*i
)
3275 if (i
->def(0).getFile() != FILE_GPR
||
3276 i
->src(0).getFile() != FILE_GPR
||
3277 i
->src(1).getFile() != FILE_GPR
||
3278 i
->src(2).getFile() != FILE_GPR
||
3279 i
->getDef(0)->reg
.data
.id
!= i
->getSrc(2)->reg
.data
.id
)
3282 // TODO: gm107 can also do this for S32, maybe other chipsets as well
3283 if (i
->dType
!= TYPE_F32
)
3286 if ((i
->src(2).mod
| Modifier(NV50_IR_MOD_NEG
)) != Modifier(NV50_IR_MOD_NEG
))
3292 if (i
->src(0).getImmediate(val
))
3294 else if (i
->src(1).getImmediate(val
))
3299 if ((i
->src(s
).mod
| Modifier(NV50_IR_MOD_NEG
)) != Modifier(NV50_IR_MOD_NEG
))
3303 i
->swapSources(0, 1);
3305 Instruction
*imm
= i
->getSrc(1)->getInsn();
3306 i
->setSrc(1, imm
->getSrc(0));
3307 if (post_ra_dead(imm
))
3308 delete_Instruction(prog
, imm
);
3312 PostRaLoadPropagation::visit(Instruction
*i
)
3317 if (prog
->getTarget()->getChipset() < 0xc0)
3318 handleMADforNV50(i
);
3320 handleMADforNVC0(i
);
3329 // =============================================================================
3331 // Common subexpression elimination. Stupid O^2 implementation.
3332 class LocalCSE
: public Pass
3335 virtual bool visit(BasicBlock
*);
3337 inline bool tryReplace(Instruction
**, Instruction
*);
3339 DLList ops
[OP_LAST
+ 1];
3342 class GlobalCSE
: public Pass
3345 virtual bool visit(BasicBlock
*);
3349 Instruction::isActionEqual(const Instruction
*that
) const
3351 if (this->op
!= that
->op
||
3352 this->dType
!= that
->dType
||
3353 this->sType
!= that
->sType
)
3355 if (this->cc
!= that
->cc
)
3358 if (this->asTex()) {
3359 if (memcmp(&this->asTex()->tex
,
3360 &that
->asTex()->tex
,
3361 sizeof(this->asTex()->tex
)))
3364 if (this->asCmp()) {
3365 if (this->asCmp()->setCond
!= that
->asCmp()->setCond
)
3368 if (this->asFlow()) {
3371 if (this->ipa
!= that
->ipa
||
3372 this->lanes
!= that
->lanes
||
3373 this->perPatch
!= that
->perPatch
)
3375 if (this->postFactor
!= that
->postFactor
)
3379 if (this->subOp
!= that
->subOp
||
3380 this->saturate
!= that
->saturate
||
3381 this->rnd
!= that
->rnd
||
3382 this->ftz
!= that
->ftz
||
3383 this->dnz
!= that
->dnz
||
3384 this->cache
!= that
->cache
||
3385 this->mask
!= that
->mask
)
3392 Instruction::isResultEqual(const Instruction
*that
) const
3396 // NOTE: location of discard only affects tex with liveOnly and quadops
3397 if (!this->defExists(0) && this->op
!= OP_DISCARD
)
3400 if (!isActionEqual(that
))
3403 if (this->predSrc
!= that
->predSrc
)
3406 for (d
= 0; this->defExists(d
); ++d
) {
3407 if (!that
->defExists(d
) ||
3408 !this->getDef(d
)->equals(that
->getDef(d
), false))
3411 if (that
->defExists(d
))
3414 for (s
= 0; this->srcExists(s
); ++s
) {
3415 if (!that
->srcExists(s
))
3417 if (this->src(s
).mod
!= that
->src(s
).mod
)
3419 if (!this->getSrc(s
)->equals(that
->getSrc(s
), true))
3422 if (that
->srcExists(s
))
3425 if (op
== OP_LOAD
|| op
== OP_VFETCH
|| op
== OP_ATOM
) {
3426 switch (src(0).getFile()) {
3427 case FILE_MEMORY_CONST
:
3428 case FILE_SHADER_INPUT
:
3430 case FILE_SHADER_OUTPUT
:
3431 return bb
->getProgram()->getType() == Program::TYPE_TESSELLATION_EVAL
;
3440 // pull through common expressions from different in-blocks
3442 GlobalCSE::visit(BasicBlock
*bb
)
3444 Instruction
*phi
, *next
, *ik
;
3447 // TODO: maybe do this with OP_UNION, too
3449 for (phi
= bb
->getPhi(); phi
&& phi
->op
== OP_PHI
; phi
= next
) {
3451 if (phi
->getSrc(0)->refCount() > 1)
3453 ik
= phi
->getSrc(0)->getInsn();
3455 continue; // probably a function input
3456 if (ik
->defCount(0xff) > 1)
3457 continue; // too painful to check if we can really push this forward
3458 for (s
= 1; phi
->srcExists(s
); ++s
) {
3459 if (phi
->getSrc(s
)->refCount() > 1)
3461 if (!phi
->getSrc(s
)->getInsn() ||
3462 !phi
->getSrc(s
)->getInsn()->isResultEqual(ik
))
3465 if (!phi
->srcExists(s
)) {
3466 Instruction
*entry
= bb
->getEntry();
3468 if (!entry
|| entry
->op
!= OP_JOIN
)
3471 bb
->insertAfter(entry
, ik
);
3472 ik
->setDef(0, phi
->getDef(0));
3473 delete_Instruction(prog
, phi
);
3481 LocalCSE::tryReplace(Instruction
**ptr
, Instruction
*i
)
3483 Instruction
*old
= *ptr
;
3485 // TODO: maybe relax this later (causes trouble with OP_UNION)
3486 if (i
->isPredicated())
3489 if (!old
->isResultEqual(i
))
3492 for (int d
= 0; old
->defExists(d
); ++d
)
3493 old
->def(d
).replace(i
->getDef(d
), false);
3494 delete_Instruction(prog
, old
);
3500 LocalCSE::visit(BasicBlock
*bb
)
3502 unsigned int replaced
;
3505 Instruction
*ir
, *next
;
3509 // will need to know the order of instructions
3511 for (ir
= bb
->getFirst(); ir
; ir
= ir
->next
)
3512 ir
->serial
= serial
++;
3514 for (ir
= bb
->getFirst(); ir
; ir
= next
) {
3521 ops
[ir
->op
].insert(ir
);
3525 for (s
= 0; ir
->srcExists(s
); ++s
)
3526 if (ir
->getSrc(s
)->asLValue())
3527 if (!src
|| ir
->getSrc(s
)->refCount() < src
->refCount())
3528 src
= ir
->getSrc(s
);
3531 for (Value::UseIterator it
= src
->uses
.begin();
3532 it
!= src
->uses
.end(); ++it
) {
3533 Instruction
*ik
= (*it
)->getInsn();
3534 if (ik
&& ik
->bb
== ir
->bb
&& ik
->serial
< ir
->serial
)
3535 if (tryReplace(&ir
, ik
))
3539 DLLIST_FOR_EACH(&ops
[ir
->op
], iter
)
3541 Instruction
*ik
= reinterpret_cast<Instruction
*>(iter
.get());
3542 if (tryReplace(&ir
, ik
))
3548 ops
[ir
->op
].insert(ir
);
3552 for (unsigned int i
= 0; i
<= OP_LAST
; ++i
)
3560 // =============================================================================
3562 // Remove computations of unused values.
3563 class DeadCodeElim
: public Pass
3566 bool buryAll(Program
*);
3569 virtual bool visit(BasicBlock
*);
3571 void checkSplitLoad(Instruction
*ld
); // for partially dead loads
3573 unsigned int deadCount
;
3577 DeadCodeElim::buryAll(Program
*prog
)
3581 if (!this->run(prog
, false, false))
3583 } while (deadCount
);
3589 DeadCodeElim::visit(BasicBlock
*bb
)
3593 for (Instruction
*i
= bb
->getExit(); i
; i
= prev
) {
3597 delete_Instruction(prog
, i
);
3599 if (i
->defExists(1) &&
3601 (i
->op
== OP_VFETCH
|| i
->op
== OP_LOAD
)) {
3604 if (i
->defExists(0) && !i
->getDef(0)->refCount()) {
3605 if (i
->op
== OP_ATOM
||
3606 i
->op
== OP_SUREDP
||
3607 i
->op
== OP_SUREDB
) {
3609 if (i
->op
== OP_ATOM
&& i
->subOp
== NV50_IR_SUBOP_ATOM_EXCH
) {
3610 i
->cache
= CACHE_CV
;
3614 } else if (i
->op
== OP_LOAD
&& i
->subOp
== NV50_IR_SUBOP_LOAD_LOCKED
) {
3615 i
->setDef(0, i
->getDef(1));
3623 // Each load can go into up to 4 destinations, any of which might potentially
3624 // be dead (i.e. a hole). These can always be split into 2 loads, independent
3625 // of where the holes are. We find the first contiguous region, put it into
3626 // the first load, and then put the second contiguous region into the second
3627 // load. There can be at most 2 contiguous regions.
3629 // Note that there are some restrictions, for example it's not possible to do
3630 // a 64-bit load that's not 64-bit aligned, so such a load has to be split
3631 // up. Also hardware doesn't support 96-bit loads, so those also have to be
3632 // split into a 64-bit and 32-bit load.
3634 DeadCodeElim::checkSplitLoad(Instruction
*ld1
)
3636 Instruction
*ld2
= NULL
; // can get at most 2 loads
3639 int32_t addr1
, addr2
;
3640 int32_t size1
, size2
;
3642 uint32_t mask
= 0xffffffff;
3644 for (d
= 0; ld1
->defExists(d
); ++d
)
3645 if (!ld1
->getDef(d
)->refCount() && ld1
->getDef(d
)->reg
.data
.id
< 0)
3647 if (mask
== 0xffffffff)
3650 addr1
= ld1
->getSrc(0)->reg
.data
.offset
;
3654 // Compute address/width for first load
3655 for (d
= 0; ld1
->defExists(d
); ++d
) {
3656 if (mask
& (1 << d
)) {
3657 if (size1
&& (addr1
& 0x7))
3659 def1
[n1
] = ld1
->getDef(d
);
3660 size1
+= def1
[n1
++]->reg
.size
;
3663 addr1
+= ld1
->getDef(d
)->reg
.size
;
3669 // Scale back the size of the first load until it can be loaded. This
3670 // typically happens for TYPE_B96 loads.
3672 !prog
->getTarget()->isAccessSupported(ld1
->getSrc(0)->reg
.file
,
3673 typeOfSize(size1
))) {
3674 size1
-= def1
[--n1
]->reg
.size
;
3678 // Compute address/width for second load
3679 for (addr2
= addr1
+ size1
; ld1
->defExists(d
); ++d
) {
3680 if (mask
& (1 << d
)) {
3681 assert(!size2
|| !(addr2
& 0x7));
3682 def2
[n2
] = ld1
->getDef(d
);
3683 size2
+= def2
[n2
++]->reg
.size
;
3686 addr2
+= ld1
->getDef(d
)->reg
.size
;
3692 // Make sure that we've processed all the values
3693 for (; ld1
->defExists(d
); ++d
)
3694 assert(!(mask
& (1 << d
)));
3696 updateLdStOffset(ld1
, addr1
, func
);
3697 ld1
->setType(typeOfSize(size1
));
3698 for (d
= 0; d
< 4; ++d
)
3699 ld1
->setDef(d
, (d
< n1
) ? def1
[d
] : NULL
);
3704 ld2
= cloneShallow(func
, ld1
);
3705 updateLdStOffset(ld2
, addr2
, func
);
3706 ld2
->setType(typeOfSize(size2
));
3707 for (d
= 0; d
< 4; ++d
)
3708 ld2
->setDef(d
, (d
< n2
) ? def2
[d
] : NULL
);
3710 ld1
->bb
->insertAfter(ld1
, ld2
);
3713 // =============================================================================
3715 #define RUN_PASS(l, n, f) \
3716 if (level >= (l)) { \
3717 if (dbgFlags & NV50_IR_DEBUG_VERBOSE) \
3718 INFO("PEEPHOLE: %s\n", #n); \
3720 if (!pass.f(this)) \
3725 Program::optimizeSSA(int level
)
3727 RUN_PASS(1, DeadCodeElim
, buryAll
);
3728 RUN_PASS(1, CopyPropagation
, run
);
3729 RUN_PASS(1, MergeSplits
, run
);
3730 RUN_PASS(2, GlobalCSE
, run
);
3731 RUN_PASS(1, LocalCSE
, run
);
3732 RUN_PASS(2, AlgebraicOpt
, run
);
3733 RUN_PASS(2, ModifierFolding
, run
); // before load propagation -> less checks
3734 RUN_PASS(1, ConstantFolding
, foldAll
);
3735 RUN_PASS(2, LateAlgebraicOpt
, run
);
3736 RUN_PASS(1, Split64BitOpPreRA
, run
);
3737 RUN_PASS(1, LoadPropagation
, run
);
3738 RUN_PASS(1, IndirectPropagation
, run
);
3739 RUN_PASS(2, MemoryOpt
, run
);
3740 RUN_PASS(2, LocalCSE
, run
);
3741 RUN_PASS(0, DeadCodeElim
, buryAll
);
3747 Program::optimizePostRA(int level
)
3749 RUN_PASS(2, FlatteningPass
, run
);
3750 RUN_PASS(2, PostRaLoadPropagation
, run
);