2

我想分析一个调用 fortran 例程的 python 代码。我使用这个 PSTATS,但如下例所示,PSTATS 没有考虑在 fortran 例程中花费的时间。

Fortran 模块 (mod_fortran.f90):

module mod_fortran
IMPLICIT NONE

CONTAINS

SUBROUTINE sub_sleep(wtime)
  INTEGER(kind = 4), intent(in) :: wtime !! in seconds
  CHARACTER(len=9)          :: cwtime

  WRITE(cwtime, 100)wtime
  write(*,*) cwtime

  CALL system(cwtime)

100 FORMAT('sleep ',I3)
END SUBROUTINE sub_sleep

end module mod_fortran

调用 fortran 模块 (test_fortran_pstats.py) 的 Python 脚本:

from mod_fortran import *
mod_fortran.sub_sleep(10)

用于打印分析统计信息的 Python 脚本 (analysis_pstats.py):

import pstats
ps = pstats.Stats('output.pstats')
ps.strip_dirs().sort_stats('tottime').print_stats(5)

教程(编译提交):

f2py -c -m mod_fortran mod_fortran.f90
time python -m profile -o output.pstats test_fortran_pstats.py
python analyse_pstats.py

结果:

$> time python -m profile -o output.pstats test_fortran_pstats.py
sleep  10
0.236u 0.080s 0:13.58 2.2%  0+0k 47280+240io 1pf+0w
$>python analyse_pstats.py
Fri Apr  3 11:51:02 2015    output.pstats
11170 function calls (10981 primitive calls) in 0.180 seconds

Pstats 只计算 0.180 秒,而 Fortran 例程在 10 秒内“休眠”。

4

0 回答 0