From 24af864d57d22375350234258e8f18fad2a08780 Mon Sep 17 00:00:00 2001
From: Kimish Patel <kimishpatel@fb.com>
Date: Tue, 20 Aug 2019 08:53:01 -0700
Subject: [PATCH] Master broken due to changed in PT API. Catching up with
 that.

Summary:
ProfiledTensorType's sizes returns optional<ListofInts>. Changes to
reflect that.

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:
---
 torch_tvm/compiler.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/torch_tvm/compiler.cpp b/torch_tvm/compiler.cpp
index e774fc3..a44bf15 100644
--- a/torch_tvm/compiler.cpp
+++ b/torch_tvm/compiler.cpp
@@ -55,7 +55,10 @@ tvm::relay::Var TVMCompiler::convertToRelay(Value* val, TVMContext ctx) {
       "Expected CPU device type but got:", device_type);
     tvm::Array<tvm::relay::IndexExpr> sizes;
     const auto& varying_sizes = pt_t->sizes();
-    for (const auto& optional_size : varying_sizes.sizes()) {
+    const auto& optional_sizes = varying_sizes.sizes();
+    TORCH_INTERNAL_ASSERT(optional_sizes);
+    const auto& pt_sizes = optional_sizes.value();
+    for (const auto& optional_size : pt_sizes) {
       TORCH_INTERNAL_ASSERT(optional_size);
       sizes.push_back(tvm::relay::IndexExpr(
             static_cast<int32_t>(optional_size.value())));