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:
177177
178178vkapi::ScalarType from_at_scalartype(c10::ScalarType at_scalartype) {{
179179 switch (at_scalartype) {{
180+ case c10::kDouble:
181+ return vkapi::kDouble;
180182 case c10::kFloat:
181183 return vkapi::kFloat;
182184 case c10::kHalf:
@@ -187,6 +189,8 @@ def generate_benchmark_fixture(self) -> str:
187189 return vkapi::kInt;
188190 case c10::kChar:
189191 return vkapi::kChar;
192+ case c10::kBool:
193+ return vkapi::kBool;
190194 default:
191195 VK_THROW("Unsupported at::ScalarType!");
192196 }}
Original file line number Diff line number Diff line change @@ -119,15 +119,11 @@ def gen_parameterization(self) -> str:
119119 case c10::kInt:
120120 return vkapi::kInt;
121121 case c10::kLong:
122- return vkapi::kLong;
123- case c10::kShort:
124- return vkapi::kShort;
122+ return vkapi::kInt;
125123 case c10::kChar:
126124 return vkapi::kChar;
127125 case c10::kBool:
128126 return vkapi::kBool;
129- case c10::kByte:
130- return vkapi::kByte;
131127 default:
132128 VK_THROW("Unsupported at::ScalarType!");
133129 }
You can’t perform that action at this time.
0 commit comments