IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added ignored pattern for Stable Diffusion Unet (#14925)

Browse Source
This commit is contained in:
Alexander Kozlov 2023-01-13 12:51:27 +04:00 committed by GitHub
parent 529f714830
commit ded9156b98
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tools/pot/openvino/tools/pot/graph/special_patterns.py
View File

@ -293,6 +293,21 @@ def create_softmax_reshape_transpose_matmul_pattern():
return pattern.set_name('softmax_reshape_transpose_matmul').pattern return pattern.set_name('softmax_reshape_transpose_matmul').pattern
# Stable diffusion UNet
@registry_ignore_patterns('blocks')
def create_stable_diffusion_pattern():
pattern = PatternBuilder()
pattern_2 = PatternBuilder()
softmax_out = pattern.append_single_op('SoftMax', 'softmax').get_last_node()
pattern_2.append_single_op('Reshape', 'reshape1')
pattern_2.append_single_op('Transpose', 'transpose')
transp_out = pattern_2.append_single_op('Reshape', 'reshape2').get_last_node()
pattern.pattern['nodes'] += pattern_2.pattern['nodes']
pattern.pattern['edges'] += pattern_2.pattern['edges']
pattern.insert_single_op([transp_out, softmax_out], None, 'MatMul', 'matmul')
return pattern.set_name('stable_diffusion').pattern
@registry_ignore_patterns('blocks') @registry_ignore_patterns('blocks')
def create_softmax_reshape_transpose_gather_matmul_pattern(): def create_softmax_reshape_transpose_gather_matmul_pattern():
pattern = PatternBuilder() pattern = PatternBuilder()