Fix load time. (pytorch#5781)

Summary:
Pull Request resolved: pytorch#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

Author: shoumikhin

extension/apple/Benchmark/Tests/Tests.mm

@@ -77,15 +77,15 @@ + (void)initialize {
                                        modelName,
                                        deviceInfo]);
         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(
@@ -107,7 +107,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);
 
           for (auto index = 0; index < num_inputs; ++index) {