Commit bb805ad
committed
Update base for Update on "[ET-VK] Allow specifying multiple storage types/memory layouts for an operator + register group norm operator"
## Changes
* Handle cases where an operator needs to specify a separate storage type / memory layout for each individual output.
## Motivation
Required for the group norm operator.
## Future Work
Currently, the `tag_memory_meta_pass` graph pass assumes that all tensors participating in a computation (aside from weights) will have the same storage type and memory layout. As more operators are being added, there are more exceptions to this rule.
The pass may need an update in the near future to make it possible to specify required storage types and memory layouts on a more granular level.
Differential Revision: [D77038781](https://our.internmc.facebook.com/intern/diff/D77038781/)
[ghstack-poisoned]File tree
112 files changed
+3685
-955
lines changed- .ci/scripts
- benchmark_tooling
- .github/workflows
- backends
- arm
- _passes
- test
- models
- ops
- vulkan/runtime/graph/ops/glsl
- xnnpack
- operators
- partition
- config
- quantizer
- runtime
- serialization
- test
- ops
- quantizer
- utils
- devtools/inspector
- numerical_comparator
- tests
- docs/source
- examples
- arm
- ethos-u-setup
- core_software
- ethos-u
- executor_runner
- demo-apps
- android/LlamaDemo
- docs/delegates
- apple_ios/LLaMA/docs/delegates
- models
- deepseek-r1-distill-llama-8B
- llama2
- llama
- config
- phi_4_mini
- qwen2_5
- qwen3
- extension
- android/executorch_android
- apple/ExecuTorch
- Exported
- __tests__
- llm/export
- test
- kernels
- optimized
- cpu
- portable
- cpu
- util
- test
- test
- runtime
- backend
- test
- core/portable_type
- c10/c10
- test/utils
Some content is hidden
Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.
112 files changed
+3685
-955
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
77 | 77 | | |
78 | 78 | | |
79 | 79 | | |
80 | | - | |
| 80 | + | |
81 | 81 | | |
82 | | - | |
| 82 | + | |
83 | 83 | | |
84 | 84 | | |
85 | 85 | | |
86 | 86 | | |
87 | 87 | | |
88 | 88 | | |
89 | | - | |
| 89 | + | |
90 | 90 | | |
91 | 91 | | |
92 | 92 | | |
| |||
97 | 97 | | |
98 | 98 | | |
99 | 99 | | |
100 | | - | |
| 100 | + | |
101 | 101 | | |
102 | 102 | | |
103 | 103 | | |
| |||
140 | 140 | | |
141 | 141 | | |
142 | 142 | | |
143 | | - | |
144 | | - | |
145 | | - | |
146 | | - | |
147 | | - | |
148 | | - | |
149 | | - | |
150 | | - | |
151 | | - | |
152 | | - | |
153 | | - | |
154 | | - | |
155 | | - | |
156 | | - | |
157 | | - | |
158 | | - | |
159 | 143 | | |
160 | 144 | | |
161 | 145 | | |
| |||
0 commit comments