Fix load time. (#5889)

Fix load time. (#5781)

Summary:
Pull Request resolved: #5781

Once loaded Module does nothing next time, we need to recreate the Module again to truly load the method again.

Reviewed By: cccclai

Differential Revision: D63678792

fbshipit-source-id: 1e8d6a379215e20620b73c05ce7f734e1d0fc5ce
(cherry picked from commit 04669a1)

Co-authored-by: Anthony Shoumikhin <[email protected]>

Author: pytorchbot

extension/apple/Benchmark/Tests/Tests.mm

@@ -54,15 +54,15 @@ + (void)initialize {
         SEL testLoadSelector = NSSelectorFromString([NSString
             stringWithFormat:@"test_load_%@_%@", directoryName, modelName]);
         IMP testLoadImplementation = imp_implementationWithBlock(^(id _self) {
-          auto __block module = std::make_unique<Module>(modelPath.UTF8String);
           [_self measureWithMetrics:@[
             [XCTClockMetric new],
             [XCTMemoryMetric new],
           ]
                             options:XCTMeasureOptions.defaultOptions
                               block:^{
-                                XCTAssertEqual(module->load_method("forward"),
-                                               Error::Ok);
+                                XCTAssertEqual(
+                                    Module(modelPath.UTF8String).load_forward(),
+                                    Error::Ok);
                               }];
         });
         class_addMethod(
@@ -81,7 +81,7 @@ + (void)initialize {
           const auto num_inputs = method_meta->num_inputs();
           XCTAssertGreaterThan(num_inputs, 0);
 
-          std::vector<TensorPtr> __block tensors;
+          std::vector<TensorPtr> tensors;
           tensors.reserve(num_inputs);
           std::vector<EValue> __block inputs;
           inputs.reserve(num_inputs);