Skip to content

Commit cea8bb0

Browse files
committed
[Docker] Upgrade oneflow to v0.8.0
The PR #15819 installed oneflow from PyPi in an attempt to unblock CI failing on the gpu docker image build. However, it seems to be a placeholder package. This PR upgrades the version of oneflow to v0.8.0 in a second attempt to unblock CI. Change-Id: I92bcc6aee79dfcbeba7c13cf0b6d91104be16f5c
1 parent 2890899 commit cea8bb0

File tree

2 files changed

+9
-4
lines changed

2 files changed

+9
-4
lines changed

docker/install/ubuntu_install_oneflow.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,4 +22,4 @@ set -o pipefail
2222

2323
pip3 install flowvision==0.1.0
2424

25-
python3 -m pip install oneflow==0.7.0
25+
python3 -m pip install oneflow==0.8.0

python/tvm/relay/frontend/oneflow.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1434,8 +1434,10 @@ def get_convert_map():
14341434
# defs/nn
14351435
"conv2d": Conv2d.get_converter(),
14361436
"deconv2d": ConvTranspose2d.get_converter(),
1437-
"maxpool_2d": MaxPool2d.get_converter(),
1438-
"avgpool_2d": AveragePool2d.get_converter(),
1437+
"max_pool_2d": MaxPool2d.get_converter(),
1438+
"avg_pool_2d": AveragePool2d.get_converter(),
1439+
"maxpool_2d": MaxPool2d.get_converter(), # Maintained for oneflow versions <= "0.7.0"
1440+
"avgpool_2d": AveragePool2d.get_converter(), # Maintained for oneflow versions <= "0.7.0"
14391441
"adaptive_avg_pool2d": AdaptiveAvgPool2d.get_converter(),
14401442
"adaptive_max_pool2d": AdaptiveMaxPool2d.get_converter(),
14411443
"dropout": Dropout.get_converter(),
@@ -1909,7 +1911,10 @@ def from_oneflow(graph, model_dir_path):
19091911
size_attr = size_str[0].replace("size=", "")
19101912
if size_attr[-2] == ",":
19111913
size_attr = size_attr.replace(",", "")
1912-
data_size = tuple(map(int, size_attr[1:-1].split(", ")))
1914+
if size_attr == "()":
1915+
data_size = ()
1916+
else:
1917+
data_size = tuple(map(int, size_attr[1:-1].split(", ")))
19131918
node_name = attrs[1]
19141919
shape[node_name] = data_size
19151920
dtype[node_name] = "float32"

0 commit comments

Comments
 (0)