Skip to content

speedup model compression UTs#956

Merged
amcadmus merged 1 commit intodeepmodeling:develfrom
denghuilu:speedup-ut
Aug 12, 2021
Merged

speedup model compression UTs#956
amcadmus merged 1 commit intodeepmodeling:develfrom
denghuilu:speedup-ut

Conversation

@denghuilu
Copy link
Member

In the benchmark on my local workstation, the model compression unit test within the $deepmd_source_dir/source/tests/test_model_compression.py consumes about 50% of the total unit test execution time. The execution time of test_model_compression.py can be largely reduced by this PR.

The main benefits of this PR are:

  • reduce the total ut time from 805.551 seconds to 422.773 seconds on my local 6-core Intel Xeon 8163 CPU environment.
  • avoid GPU memory conflict during the os.system calls, check this.

@denghuilu denghuilu requested review from amcadmus and njzjz August 12, 2021 10:49
@codecov-commenter
Copy link

codecov-commenter commented Aug 12, 2021

Codecov Report

Merging #956 (0e2e79b) into devel (3c2a2bd) will decrease coverage by 0.10%.
The diff coverage is n/a.

Impacted file tree graph

@@            Coverage Diff             @@
##            devel     #956      +/-   ##
==========================================
- Coverage   75.51%   75.41%   -0.11%     
==========================================
  Files          85       85              
  Lines        6796     6796              
==========================================
- Hits         5132     5125       -7     
- Misses       1664     1671       +7     
Impacted Files Coverage Δ
deepmd/utils/tabulate.py 86.18% <0.00%> (-1.66%) ⬇️
deepmd/descriptor/se_a.py 96.18% <0.00%> (-1.39%) ⬇️

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 3c2a2bd...0e2e79b. Read the comment docs.

@amcadmus amcadmus merged commit 074f3f2 into deepmodeling:devel Aug 12, 2021
gzq942560379 pushed a commit to HPC-AI-Team/deepmd-kit that referenced this pull request Sep 2, 2021
njzjz added a commit to njzjz/deepmd-kit that referenced this pull request Jul 15, 2023
I found that deepmodeling#956 broke this unit test. It only tested a normal model.

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
wanghan-iapcm pushed a commit that referenced this pull request Jul 15, 2023
I found that #956 broke this unit test, making it only testing a
standard model.

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants