diff options
| -rw-r--r-- | src/compiler/nir/nir.h | 1 | ||||
| -rw-r--r-- | src/compiler/nir/nir_lower_double_ops.c | 14 |
2 files changed, 15 insertions, 0 deletions
diff --git a/src/compiler/nir/nir.h b/src/compiler/nir/nir.h index 252567ceae5..ac96727cad6 100644 --- a/src/compiler/nir/nir.h +++ b/src/compiler/nir/nir.h @@ -2420,6 +2420,7 @@ typedef enum { nir_lower_dtrunc = (1 << 3), nir_lower_dfloor = (1 << 4), nir_lower_dceil = (1 << 5), + nir_lower_dfract = (1 << 6) } nir_lower_doubles_options; void nir_lower_doubles(nir_shader *shader, nir_lower_doubles_options options); diff --git a/src/compiler/nir/nir_lower_double_ops.c b/src/compiler/nir/nir_lower_double_ops.c index 2d94f78ec4e..f1fa2c3fd2d 100644 --- a/src/compiler/nir/nir_lower_double_ops.c +++ b/src/compiler/nir/nir_lower_double_ops.c @@ -383,6 +383,12 @@ lower_ceil(nir_builder *b, nir_ssa_def *src) nir_fadd(b, tr, nir_imm_double(b, 1.0))); } +static nir_ssa_def * +lower_fract(nir_builder *b, nir_ssa_def *src) +{ + return nir_fsub(b, src, nir_ffloor(b, src)); +} + static void lower_doubles_instr(nir_alu_instr *instr, nir_lower_doubles_options options) { @@ -421,6 +427,11 @@ lower_doubles_instr(nir_alu_instr *instr, nir_lower_doubles_options options) return; break; + case nir_op_ffract: + if (!(options & nir_lower_dfract)) + return; + break; + default: return; } @@ -453,6 +464,9 @@ lower_doubles_instr(nir_alu_instr *instr, nir_lower_doubles_options options) case nir_op_fceil: result = lower_ceil(&bld, src); break; + case nir_op_ffract: + result = lower_fract(&bld, src); + break; default: unreachable("unhandled opcode"); } |