Fix input setting and invocation.

Summary: . Reviewed By: kirklandsign Differential Revision: D63640035
pytorch · Sep 30, 2024 · 41f3516 · 41f3516
1 parent 4bf7e2f
commit 41f3516
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/extension/apple/Benchmark/Tests/Tests.mm b/extension/apple/Benchmark/Tests/Tests.mm
@@ -122,7 +122,8 @@ + (void)initialize {
               const auto sizes = tensor_meta->sizes();
               tensors.emplace_back(ones({sizes.begin(), sizes.end()},
                                         tensor_meta->scalar_type()));
-              module.set_input(tensors.back(), index);
+              XCTAssertEqual(module->set_input(tensors.back(), index),
+                             Error::Ok);
             } break;
             default:
               XCTFail("Unsupported tag %i at input %d", *input_tag, index);
@@ -134,7 +135,7 @@ + (void)initialize {
           ]
                             options:XCTMeasureOptions.defaultOptions
                               block:^{
-                                XCTAssertEqual(module->forward(inputs).error(),
+                                XCTAssertEqual(module->forward().error(),
                                                Error::Ok);
                               }];
         });