提交 937e3e81 编写于 作者: L luxuhui

support ndk16 and python3, fix ci error

N/A
Signed-off-by: NLuxuhui <luxuhui@xiaomi.com>
上级 adace180
...@@ -48,7 +48,7 @@ inline unsigned int GetOpenclQueueWindowSizeFromEnv() { ...@@ -48,7 +48,7 @@ inline unsigned int GetOpenclQueueWindowSizeFromEnv() {
GetEnv(kOpenClWindowSize, &str_size); GetEnv(kOpenClWindowSize, &str_size);
unsigned int window_size = 0; unsigned int window_size = 0;
if (str_size.size() > 0) { if (str_size.size() > 0) {
window_size = std::stoi(str_size); window_size = atoi(str_size.c_str());
} }
return window_size; return window_size;
} }
......
...@@ -202,7 +202,7 @@ void TestMVNormRandom(const std::vector<index_t> &input_shape, ...@@ -202,7 +202,7 @@ void TestMVNormRandom(const std::vector<index_t> &input_shape,
net.RunOp(D); net.RunOp(D);
if (DataTypeToEnum<T>::value == DT_HALF) { if (DataTypeToEnum<T>::value == DT_HALF) {
ExpectTensorNear<float>(*expected, *net.GetOutput("Output"), 1e-2, 1e-3); ExpectTensorNear<float>(*expected, *net.GetOutput("Output"), 1e-2, 1e-2);
} else { } else {
ExpectTensorNear<float>(*expected, *net.GetOutput("Output"), 1e-5); ExpectTensorNear<float>(*expected, *net.GetOutput("Output"), 1e-5);
} }
......
...@@ -1353,7 +1353,7 @@ class Transformer(base_converter.ConverterInterface): ...@@ -1353,7 +1353,7 @@ class Transformer(base_converter.ConverterInterface):
visited = set() visited = set()
sorted_nodes = [] sorted_nodes = []
output_nodes = self._option.check_nodes.keys() output_nodes = list(self._option.check_nodes.keys())
if not self._quantize_activation_info: if not self._quantize_activation_info:
output_nodes.extend(self._option.output_nodes) output_nodes.extend(self._option.output_nodes)
for output_node in output_nodes: for output_node in output_nodes:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册