We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 451ca72 commit 7b6eeb8Copy full SHA for 7b6eeb8
extension/runner_util/inputs.cpp
@@ -23,6 +23,21 @@ Result<BufferCleanup> prepare_input_tensors(Method& method) {
23
void** inputs = (void**)malloc(num_inputs * sizeof(void*));
24
25
for (size_t i = 0; i < num_inputs; i++) {
26
+ auto tag = method_meta.input_tag(i);
27
+ if (!tag.ok()) {
28
+ return tag.error();
29
+ }
30
+
31
+ // Skip valid non-tensor inputs such as ints & floats.
32
+ switch (tag.get()) {
33
+ case Tag::Int:
34
+ case Tag::Double:
35
+ case Tag::Bool:
36
+ continue;
37
+ default:
38
+ break;
39
40
41
Result<TensorInfo> tensor_meta = method_meta.input_tensor_meta(i);
42
if (!tensor_meta.ok()) {
43
ET_LOG(Info, "Skipping non-tensor input %zu", i);
0 commit comments