Merge pull request #142 from tenstorrent/nvukobrat/resnet

[Model] Placeholder for ResNet 50 (initial planing & stats)

pybuda/test/mlir/resnet/test_resnet_inference.py

@@ -0,0 +1,27 @@
+# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
+
+# SPDX-License-Identifier: Apache-2.0
+
+import torch
+from torchvision.models.resnet import resnet50
+
+import pybuda
+
+
+def test_resnet_inference():
+    # Compiler configurations
+    compiler_cfg = pybuda.config._get_global_compiler_config()
+    compiler_cfg.enable_tvm_cpu_fallback = False
+
+    # Load ResNet50 model
+    framework_model = resnet50()
+    framework_model.eval()
+
+    input_image = torch.rand(1, 3, 224, 224)
+
+    # Sanity run
+    generation_output = framework_model(input_image)
+    print(generation_output)
+
+    # Compile the model
+    compiled_model = pybuda.compile(framework_model, input_image)