101e04c3fSmrg/*
201e04c3fSmrg * Copyright © 2014 Intel Corporation
301e04c3fSmrg *
401e04c3fSmrg * Permission is hereby granted, free of charge, to any person obtaining a
501e04c3fSmrg * copy of this software and associated documentation files (the "Software"),
601e04c3fSmrg * to deal in the Software without restriction, including without limitation
701e04c3fSmrg * the rights to use, copy, modify, merge, publish, distribute, sublicense,
801e04c3fSmrg * and/or sell copies of the Software, and to permit persons to whom the
901e04c3fSmrg * Software is furnished to do so, subject to the following conditions:
1001e04c3fSmrg *
1101e04c3fSmrg * The above copyright notice and this permission notice (including the next
1201e04c3fSmrg * paragraph) shall be included in all copies or substantial portions of the
1301e04c3fSmrg * Software.
1401e04c3fSmrg *
1501e04c3fSmrg * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
1601e04c3fSmrg * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
1701e04c3fSmrg * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
1801e04c3fSmrg * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
1901e04c3fSmrg * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
2001e04c3fSmrg * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
2101e04c3fSmrg * IN THE SOFTWARE.
2201e04c3fSmrg *
2301e04c3fSmrg * Authors:
2401e04c3fSmrg *    Jason Ekstrand (jason@jlekstrand.net)
2501e04c3fSmrg *    Connor Abbott (cwabbott0@gmail.com)
2601e04c3fSmrg *
2701e04c3fSmrg */
2801e04c3fSmrg
2901e04c3fSmrg#include "nir_instr_set.h"
3001e04c3fSmrg
3101e04c3fSmrg/*
3201e04c3fSmrg * Implements common subexpression elimination
3301e04c3fSmrg */
3401e04c3fSmrg
3501e04c3fSmrgstatic bool
367ec681f3Smrgdominates(const nir_instr *old_instr, const nir_instr *new_instr)
3701e04c3fSmrg{
387ec681f3Smrg   return nir_block_dominates(old_instr->block, new_instr->block);
3901e04c3fSmrg}
4001e04c3fSmrg
4101e04c3fSmrgstatic bool
4201e04c3fSmrgnir_opt_cse_impl(nir_function_impl *impl)
4301e04c3fSmrg{
4401e04c3fSmrg   struct set *instr_set = nir_instr_set_create(NULL);
4501e04c3fSmrg
467ec681f3Smrg   _mesa_set_resize(instr_set, impl->ssa_alloc);
477ec681f3Smrg
4801e04c3fSmrg   nir_metadata_require(impl, nir_metadata_dominance);
4901e04c3fSmrg
507ec681f3Smrg   bool progress = false;
517ec681f3Smrg   nir_foreach_block(block, impl) {
527ec681f3Smrg      nir_foreach_instr_safe(instr, block)
537ec681f3Smrg         progress |= nir_instr_set_add_or_rewrite(instr_set, instr, dominates);
547ec681f3Smrg   }
5501e04c3fSmrg
567e102996Smaya   if (progress) {
5701e04c3fSmrg      nir_metadata_preserve(impl, nir_metadata_block_index |
5801e04c3fSmrg                                  nir_metadata_dominance);
597e102996Smaya   } else {
607ec681f3Smrg      nir_metadata_preserve(impl, nir_metadata_all);
617e102996Smaya   }
6201e04c3fSmrg
6301e04c3fSmrg   nir_instr_set_destroy(instr_set);
6401e04c3fSmrg   return progress;
6501e04c3fSmrg}
6601e04c3fSmrg
6701e04c3fSmrgbool
6801e04c3fSmrgnir_opt_cse(nir_shader *shader)
6901e04c3fSmrg{
7001e04c3fSmrg   bool progress = false;
7101e04c3fSmrg
7201e04c3fSmrg   nir_foreach_function(function, shader) {
7301e04c3fSmrg      if (function->impl)
7401e04c3fSmrg         progress |= nir_opt_cse_impl(function->impl);
7501e04c3fSmrg   }
7601e04c3fSmrg
7701e04c3fSmrg   return progress;
7801e04c3fSmrg}
7901e04c3fSmrg
80