Skip to content

Commit 7a1f7d0

Browse files
authored
[Docker] Upgrade oneflow to v0.8.0 (#15862)
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 b8abff9 commit 7a1f7d0

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
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 & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -742,7 +742,6 @@ class ExpandDim(OneFlowOpConverter):
742742

743743
@classmethod
744744
def _impl_v1(cls, inputs, attrs, params):
745-
746745
return _op.expand_dims(inputs[0], axis=attrs.get("axis", 0))
747746

748747

@@ -1434,8 +1433,10 @@ def get_convert_map():
14341433
# defs/nn
14351434
"conv2d": Conv2d.get_converter(),
14361435
"deconv2d": ConvTranspose2d.get_converter(),
1437-
"maxpool_2d": MaxPool2d.get_converter(),
1438-
"avgpool_2d": AveragePool2d.get_converter(),
1436+
"max_pool_2d": MaxPool2d.get_converter(),
1437+
"avg_pool_2d": AveragePool2d.get_converter(),
1438+
"maxpool_2d": MaxPool2d.get_converter(), # Maintained for oneflow versions <= "0.7.0"
1439+
"avgpool_2d": AveragePool2d.get_converter(), # Maintained for oneflow versions <= "0.7.0"
14391440
"adaptive_avg_pool2d": AdaptiveAvgPool2d.get_converter(),
14401441
"adaptive_max_pool2d": AdaptiveMaxPool2d.get_converter(),
14411442
"dropout": Dropout.get_converter(),
@@ -1909,7 +1910,10 @@ def from_oneflow(graph, model_dir_path):
19091910
size_attr = size_str[0].replace("size=", "")
19101911
if size_attr[-2] == ",":
19111912
size_attr = size_attr.replace(",", "")
1912-
data_size = tuple(map(int, size_attr[1:-1].split(", ")))
1913+
if size_attr == "()":
1914+
data_size = ()
1915+
else:
1916+
data_size = tuple(map(int, size_attr[1:-1].split(", ")))
19131917
node_name = attrs[1]
19141918
shape[node_name] = data_size
19151919
dtype[node_name] = "float32"

0 commit comments

Comments
 (0)