diff --git a/mace/utils/tuner.h b/mace/utils/tuner.h index da796ec08fa758cf161565c39d0f3e3373985aa7..f7c09769977e957a3908b1ce1d64b4dd72b265ba 100644 --- a/mace/utils/tuner.h +++ b/mace/utils/tuner.h @@ -48,7 +48,7 @@ inline unsigned int GetOpenclQueueWindowSizeFromEnv() { GetEnv(kOpenClWindowSize, &str_size); unsigned int window_size = 0; if (str_size.size() > 0) { - window_size = std::stoi(str_size); + window_size = atoi(str_size.c_str()); } return window_size; } diff --git a/test/ccunit/mace/ops/mvnorm_test.cc b/test/ccunit/mace/ops/mvnorm_test.cc index 3bafaa1b714492e9ce717aba33ee54fe44681830..84159049cb87324ca15aa91a9e3e828ff62562f5 100644 --- a/test/ccunit/mace/ops/mvnorm_test.cc +++ b/test/ccunit/mace/ops/mvnorm_test.cc @@ -202,7 +202,7 @@ void TestMVNormRandom(const std::vector &input_shape, net.RunOp(D); if (DataTypeToEnum::value == DT_HALF) { - ExpectTensorNear(*expected, *net.GetOutput("Output"), 1e-2, 1e-3); + ExpectTensorNear(*expected, *net.GetOutput("Output"), 1e-2, 1e-2); } else { ExpectTensorNear(*expected, *net.GetOutput("Output"), 1e-5); } diff --git a/tools/python/transform/transformer.py b/tools/python/transform/transformer.py index bb9154f8e09bdfc7b83ac2aa3dfd8cb775fc6478..bbf14b4f56617a475d2eebc5395f7ef04f0f9dee 100644 --- a/tools/python/transform/transformer.py +++ b/tools/python/transform/transformer.py @@ -1353,7 +1353,7 @@ class Transformer(base_converter.ConverterInterface): visited = set() sorted_nodes = [] - output_nodes = self._option.check_nodes.keys() + output_nodes = list(self._option.check_nodes.keys()) if not self._quantize_activation_info: output_nodes.extend(self._option.output_nodes) for output_node in output_nodes: