FilesExpand file tree

 master

/

timer.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

57 lines (44 loc) · 1.19 KB

 master

/

timer.py

Top

File metadata and controls

• Code
• Blame

57 lines (44 loc) · 1.19 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

#######################################################

# Copyright (c) 2015, ArrayFire

# All rights reserved.

#

# This file is distributed under 3-clause BSD license.

# The complete license agreement can be obtained at:

# http://arrayfire.com/licenses/BSD-3-Clause

########################################################

"""

Functions to time arrayfire.

"""

from .library import *

from .device import (sync, eval)

from time import time

import math

def timeit(af_func, *args):

"""

Function to time arrayfire functions.

Parameters

----------

af_func : arrayfire function

*args : arguments to `af_func`

Returns

--------

t : Time in seconds

"""

sample_trials = 3

sample_time = 1E20

for i in range(sample_trials):

start = time()

res = af_func(*args)

eval(res)

sync()

sample_time = min(sample_time, time() - start)

if (sample_time >= 0.5):

return sample_time

num_iters = max(math.ceil(1.0 / sample_time), 3.0)

start = time()

for i in range(int(num_iters)):

res = af_func(*args)

eval(res)

sync()

sample_time = (time() - start) / num_iters

return sample_time