"docs/zh_cn/vscode:/vscode.git/clone" did not exist on "05bbce8b0859a80a72ba29d35564d123eaea9ded"
Commit 237371ad authored by Anthony Chang's avatar Anthony Chang
Browse files

prevent integer overflow

parent 047cee2b
...@@ -268,7 +268,7 @@ int main(int argc, char* argv[]) ...@@ -268,7 +268,7 @@ int main(int argc, char* argv[])
float ave_time = invoker.Run(argument, StreamConfig{nullptr, time_kernel}); float ave_time = invoker.Run(argument, StreamConfig{nullptr, time_kernel});
std::size_t flop = std::size_t(2) * (M * N * K + M * N * O); std::size_t flop = (size_t)M * N * K * 2 + (size_t)M * N * O * 2;
std::size_t num_btype = sizeof(ADataType) * M * K + sizeof(B0DataType) * K * N + std::size_t num_btype = sizeof(ADataType) * M * K + sizeof(B0DataType) * K * N +
sizeof(B1DataType) * N * O + sizeof(CDataType) * M * O; sizeof(B1DataType) * N * O + sizeof(CDataType) * M * O;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment