From 1604ebaf10189c3558e5182990e8d5dc41adc1dd Mon Sep 17 00:00:00 2001 From: Haicheng Wu <57973641+hwu36@users.noreply.github.com> Date: Sun, 8 May 2022 21:47:15 -0400 Subject: [PATCH] Update generator.py stop generating analytical conv kernels to reduce kernel number --- tools/library/scripts/generator.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tools/library/scripts/generator.py b/tools/library/scripts/generator.py index 1f2301f1..bc03f2ef 100644 --- a/tools/library/scripts/generator.py +++ b/tools/library/scripts/generator.py @@ -344,8 +344,9 @@ def CreateConv2dOperator(manifest, layout, tile_descriptions, data_type, alignme # one exceptional case # iterator algorithm (analytic and optimized) - iterator_algorithms = [IteratorAlgorithm.Analytic, IteratorAlgorithm.Optimized] - + #iterator_algorithms = [IteratorAlgorithm.Analytic, IteratorAlgorithm.Optimized] + iterator_algorithms = [IteratorAlgorithm.Optimized] + # by default, only generate the largest tile size, largest alignment, and optimized iterator if manifest.kernel_filter == '': tile_descriptions = [tile_descriptions[0],] @@ -526,7 +527,8 @@ def CreateConv3dOperator(manifest, layout, tile_descriptions, data_type, alignme alignment_c = min(8, alignment) # iterator algorithm (analytic and optimized) - iterator_algorithms = [IteratorAlgorithm.Analytic, IteratorAlgorithm.Optimized] + #iterator_algorithms = [IteratorAlgorithm.Analytic, IteratorAlgorithm.Optimized] + iterator_algorithms = [IteratorAlgorithm.Optimized] # by default, only generate the largest tile size and optimized iterators if manifest.kernel_filter == '':