File tree Expand file tree Collapse file tree 2 files changed +5
-5
lines changed
backends/vulkan/test/op_tests/utils Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -177,6 +177,8 @@ def generate_benchmark_fixture(self) -> str:
177
177
178
178
vkapi::ScalarType from_at_scalartype(c10::ScalarType at_scalartype) {{
179
179
switch (at_scalartype) {{
180
+ case c10::kDouble:
181
+ return vkapi::kDouble;
180
182
case c10::kFloat:
181
183
return vkapi::kFloat;
182
184
case c10::kHalf:
@@ -187,6 +189,8 @@ def generate_benchmark_fixture(self) -> str:
187
189
return vkapi::kInt;
188
190
case c10::kChar:
189
191
return vkapi::kChar;
192
+ case c10::kBool:
193
+ return vkapi::kBool;
190
194
default:
191
195
VK_THROW("Unsupported at::ScalarType!");
192
196
}}
Original file line number Diff line number Diff line change @@ -119,15 +119,11 @@ def gen_parameterization(self) -> str:
119
119
case c10::kInt:
120
120
return vkapi::kInt;
121
121
case c10::kLong:
122
- return vkapi::kLong;
123
- case c10::kShort:
124
- return vkapi::kShort;
122
+ return vkapi::kInt;
125
123
case c10::kChar:
126
124
return vkapi::kChar;
127
125
case c10::kBool:
128
126
return vkapi::kBool;
129
- case c10::kByte:
130
- return vkapi::kByte;
131
127
default:
132
128
VK_THROW("Unsupported at::ScalarType!");
133
129
}
You can’t perform that action at this time.
0 commit comments