mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-10 06:17:55 -05:00
795fc33001
* Update default CPU compilation flags.c5a6cdc8dd52eb7e9b82tweak CPU iree-compile flags to match upstream changes. * Add an option for data tiling on SD models.
89 lines
3.6 KiB
JSON
89 lines
3.6 KiB
JSON
{
|
|
"unet": {
|
|
"tuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": []
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": []
|
|
}
|
|
},
|
|
"untuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": [
|
|
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-global-opt-detach-elementwise-from-named-ops,iree-global-opt-convert-1x1-filter-conv2d-to-matmul,iree-preprocessing-convert-conv2d-to-img2col,iree-preprocessing-pad-linalg-ops{pad-size=32}))"
|
|
]
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": [
|
|
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-global-opt-detach-elementwise-from-named-ops,iree-global-opt-convert-1x1-filter-conv2d-to-matmul,iree-preprocessing-convert-conv2d-to-img2col,iree-preprocessing-pad-linalg-ops{pad-size=16}))"
|
|
]
|
|
}
|
|
}
|
|
},
|
|
"vae": {
|
|
"tuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": [],
|
|
"specified_compilation_flags": {
|
|
"cuda": [],
|
|
"default_device": [
|
|
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-global-opt-detach-elementwise-from-named-ops,iree-global-opt-convert-1x1-filter-conv2d-to-matmul,iree-preprocessing-convert-conv2d-to-img2col,iree-preprocessing-pad-linalg-ops{pad-size=32},iree-linalg-ext-convert-conv2d-to-winograd))"
|
|
]
|
|
}
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": [],
|
|
"specified_compilation_flags": {
|
|
"cuda": [],
|
|
"default_device": [
|
|
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-global-opt-detach-elementwise-from-named-ops,iree-global-opt-convert-1x1-filter-conv2d-to-matmul,iree-preprocessing-convert-conv2d-to-img2col,iree-preprocessing-pad-linalg-ops{pad-size=16},iree-linalg-ext-convert-conv2d-to-winograd))"
|
|
]
|
|
}
|
|
}
|
|
},
|
|
"untuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": [
|
|
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-global-opt-detach-elementwise-from-named-ops,iree-global-opt-convert-1x1-filter-conv2d-to-matmul,iree-preprocessing-convert-conv2d-to-img2col,iree-preprocessing-pad-linalg-ops{pad-size=32},iree-linalg-ext-convert-conv2d-to-winograd))"
|
|
]
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": [
|
|
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-global-opt-detach-elementwise-from-named-ops,iree-global-opt-convert-1x1-filter-conv2d-to-matmul,iree-preprocessing-convert-conv2d-to-img2col,iree-preprocessing-pad-linalg-ops{pad-size=16},iree-linalg-ext-convert-conv2d-to-winograd))"
|
|
]
|
|
}
|
|
}
|
|
},
|
|
"clip": {
|
|
"tuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": [
|
|
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-preprocessing-pad-linalg-ops{pad-size=16}))",
|
|
"--iree-opt-data-tiling=False"
|
|
]
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": [
|
|
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-preprocessing-pad-linalg-ops{pad-size=16}))",
|
|
"--iree-opt-data-tiling=False"
|
|
]
|
|
}
|
|
},
|
|
"untuned": {
|
|
"fp16": {
|
|
"default_compilation_flags": [
|
|
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-preprocessing-pad-linalg-ops{pad-size=16}))",
|
|
"--iree-opt-data-tiling=False"
|
|
]
|
|
},
|
|
"fp32": {
|
|
"default_compilation_flags": [
|
|
"--iree-preprocessing-pass-pipeline=builtin.module(func.func(iree-preprocessing-pad-linalg-ops{pad-size=16}))",
|
|
"--iree-opt-data-tiling=False"
|
|
]
|
|
}
|
|
}
|
|
}
|
|
}
|