未验证 提交 22e3d85d 编写于 作者: L Liu Qi 提交者: GitHub

Merge branch 'master' into master

...@@ -6,7 +6,7 @@ about: Bug and Feature Report ...@@ -6,7 +6,7 @@ about: Bug and Feature Report
Before you open an issue, please make sure you have tried the following steps: Before you open an issue, please make sure you have tried the following steps:
1. Make sure your ENV is the same with (https://mace.readthedocs.io/en/latest/installation/env_requirement.html). 1. Make sure your **environment** is the same with (https://mace.readthedocs.io/en/latest/installation/env_requirement.html).
2. Have you ever read the document for your usage? 2. Have you ever read the document for your usage?
3. The form below must be filled. 3. The form below must be filled.
...@@ -14,10 +14,11 @@ Before you open an issue, please make sure you have tried the following steps: ...@@ -14,10 +14,11 @@ Before you open an issue, please make sure you have tried the following steps:
### System information ### System information
- **OS Platform and Distribution (e.g., Linux Ubuntu 16.04)**: - **OS Platform and Distribution (e.g., Linux Ubuntu 16.04)**:
- **NDK version(e.g. 15c)**: - **NDK version(e.g., 15c)**:
- **GCC version(if compiling for host, e.g., 5.4.0)**:
- **MACE version (Use the command: git describe --long --tags)**: - **MACE version (Use the command: git describe --long --tags)**:
- **Python version(2.7)**: - **Python version(2.7)**:
- **Bazel version (e.g. 0.13.0)**: - **Bazel version (e.g., 0.13.0)**:
### Model deploy file (*.yml) ### Model deploy file (*.yml)
```yaml ```yaml
...@@ -28,16 +29,17 @@ Before you open an issue, please make sure you have tried the following steps: ...@@ -28,16 +29,17 @@ Before you open an issue, please make sure you have tried the following steps:
A clear and concise description of what the bug is. A clear and concise description of what the bug is.
### To Reproduce ### To Reproduce
Steps to reproduce the behavior: Steps to reproduce the problem:
```bash ```bash
1. cd /path/to/mace 1. cd /path/to/mace
2. python tools/converter.py convert --config_file=/path/to/your/model_deployment_file 2. python tools/converter.py convert --config_file=/path/to/your/model_deployment_file
``` ```
### Error information / logs ### Error information / logs
Please include the **full** log and/or traceback here.
```bash ```bash
LOGs LOGs
``` ```
### Additional context ### Additional context
Add any other context about the problem here. Add any other context about the problem here, e.g., what you have modified about the code.
...@@ -38,7 +38,7 @@ std::vector<uint32_t> LocalWS(const uint32_t *gws, const uint32_t kwg_size) { ...@@ -38,7 +38,7 @@ std::vector<uint32_t> LocalWS(const uint32_t *gws, const uint32_t kwg_size) {
kwg_size / lws[1]); kwg_size / lws[1]);
} }
} }
lws[0] = std::max<uint32_t>(lws[0], 1); lws[0] = std::max<uint32_t>(std::min<uint32_t>(lws[0], kwg_size / lws[1]), 1);
const uint32_t lws_size = lws[0] * lws[1]; const uint32_t lws_size = lws[0] * lws[1];
lws[2] = std::min<uint32_t>((cache_size / kernel_cache_size / lws_size) * 4, lws[2] = std::min<uint32_t>((cache_size / kernel_cache_size / lws_size) * 4,
gws[2]); gws[2]);
......
...@@ -92,7 +92,8 @@ genrule( ...@@ -92,7 +92,8 @@ genrule(
"$@ " + "$@ " +
"$$tmp_mri_file);" + "$$tmp_mri_file);" +
"$(AR) -M <$$tmp_mri_file;" + "$(AR) -M <$$tmp_mri_file;" +
"rm -rf $$tmp_mri_file;", "rm -rf $$tmp_mri_file;" +
"$(STRIP) -x $@;",
tools = ["//mace/python/tools:archive_static_lib"], tools = ["//mace/python/tools:archive_static_lib"],
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
) )
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册