Skip to content

Instantly share code, notes, and snippets.

@bast
Created April 6, 2021 18:36
Show Gist options
  • Save bast/1a85d6aaf2805de6c31e1742c1e57772 to your computer and use it in GitHub Desktop.
Save bast/1a85d6aaf2805de6c31e1742c1e57772 to your computer and use it in GitHub Desktop.
import cProfile
from dftd3.dftd3 import D3Configuration, d3
def main(config, charges, coordinates):
d3_au = d3(config, charges, *coordinates)
coordinates = [
4.01376695,
-0.42094027,
0.00413283,
2.37659705,
-2.06988695,
-0.07118787,
3.23930049,
2.02346393,
0.03394893,
6.09063455,
-0.5931548,
0.05363799,
1.40721479,
2.12118734,
-0.0108338,
-4.04214497,
0.34532288,
-0.03309288,
-2.55773675,
2.13450991,
-0.01370996,
-3.05560778,
-2.01961834,
0.06482139,
-6.12485749,
0.3304205,
-0.12008265,
-1.22314224,
-1.95250094,
0.14005516,
]
charges = [6, 8, 8, 1, 1, 6, 8, 8, 1, 1]
functional = "B3LYP"
damping = "zero"
#damping = "bj"
config = D3Configuration(functional=functional, damp=damping)
cProfile.run("main(config, charges, coordinates)")
@robertodr
Copy link

Note for posterity: this will work with master for this repo https://github.com/Rafael-G-C/pyDFTD3

@robertodr
Copy link

Result of running the script:

>>> D3-dispersion correction with zero damping
    - Damping parameters (from database): s6 = 1.0; rs6 = 1.261; s8 = 1.703
    - 3-body term will not be calculated

WARNING:absl:No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
         1777087 function calls (1765409 primitive calls) in 1.538 seconds

   Ordered by: standard name

   ncalls  tottime  percall  cumtime  percall filename:lineno(function)
    31525    0.015    0.000    0.064    0.000 <__array_function__ internals>:2(result_type)
    14470    0.007    0.000    0.028    0.000 <__array_function__ internals>:2(shape)
        1    0.000    0.000    1.538    1.538 <string>:1(<module>)
       17    0.000    0.000    0.000    0.000 <string>:1(__new__)
        1    0.000    0.000    0.000    0.000 __init__.py:1036(findCaller)
        1    0.000    0.000    0.000    0.000 __init__.py:1047(__init__)
        1    0.000    0.000    0.000    0.000 __init__.py:1058(flush)
        1    0.000    0.000    0.000    0.000 __init__.py:1069(emit)
        7    0.000    0.000    0.000    0.000 __init__.py:1112(log)
        1    0.000    0.000    0.000    0.000 __init__.py:1131(handle)
        1    0.000    0.000    0.000    0.000 __init__.py:119(getLevelName)
        7    0.000    0.000    0.000    0.000 __init__.py:1485(log)
        1    0.000    0.000    0.000    0.000 __init__.py:1538(makeRecord)
        1    0.000    0.000    0.000    0.000 __init__.py:1553(_log)
        1    0.000    0.000    0.000    0.000 __init__.py:1589(addHandler)
        1    0.000    0.000    0.000    0.000 __init__.py:1633(callHandlers)
        2    0.000    0.000    0.000    0.000 __init__.py:1663(getEffectiveLevel)
        7    0.000    0.000    0.000    0.000 __init__.py:1677(isEnabledFor)
        1    0.000    0.000    0.000    0.000 __init__.py:189(_checkLevel)
        1    0.000    0.000    0.000    0.000 __init__.py:1896(basicConfig)
        6    0.000    0.000    0.000    0.000 __init__.py:214(_acquireLock)
        6    0.000    0.000    0.000    0.000 __init__.py:223(_releaseLock)
        1    0.000    0.000    0.000    0.000 __init__.py:244(_register_at_fork_reinit_lock)
        1    0.000    0.000    0.000    0.000 __init__.py:284(__init__)
        1    0.000    0.000    0.000    0.000 __init__.py:360(getMessage)
        1    0.000    0.000    0.000    0.000 __init__.py:391(warning)
        1    0.000    0.000    0.000    0.000 __init__.py:420(__init__)
        1    0.000    0.000    0.000    0.000 __init__.py:423(usesTime)
        1    0.000    0.000    0.000    0.000 __init__.py:426(validate)
        1    0.000    0.000    0.000    0.000 __init__.py:431(_format)
        1    0.000    0.000    0.000    0.000 __init__.py:434(format)
        7    0.000    0.000    0.000    0.000 __init__.py:525(log)
        1    0.000    0.000    0.000    0.000 __init__.py:555(__init__)
        6    0.000    0.000    0.000    0.000 __init__.py:559(vlog)
        1    0.000    0.000    0.000    0.000 __init__.py:629(usesTime)
        1    0.000    0.000    0.000    0.000 __init__.py:635(formatMessage)
        1    0.000    0.000    0.000    0.000 __init__.py:651(format)
        1    0.000    0.000    0.000    0.000 __init__.py:772(__init__)
        2    0.000    0.000    0.000    0.000 __init__.py:792(filter)
        1    0.000    0.000    0.000    0.000 __init__.py:839(_addHandlerRef)
        1    0.000    0.000    0.000    0.000 __init__.py:858(__init__)
        1    0.000    0.000    0.000    0.000 __init__.py:887(createLock)
        2    0.000    0.000    0.000    0.000 __init__.py:894(acquire)
        2    0.000    0.000    0.000    0.000 __init__.py:901(release)
        1    0.000    0.000    0.000    0.000 __init__.py:914(format)
        1    0.000    0.000    0.000    0.000 __init__.py:937(handle)
        1    0.000    0.000    0.000    0.000 __init__.py:955(setFormatter)
     7294    0.003    0.000    0.013    0.000 _asarray.py:23(asarray)
        3    0.000    0.000    0.000    0.000 _collections_abc.py:657(get)
        2    0.000    0.000    0.000    0.000 _dtype.py:24(_kind_name)
        2    0.000    0.000    0.000    0.000 _dtype.py:307(_name_includes_bit_suffix)
        2    0.000    0.000    0.000    0.000 _dtype.py:321(_name_get)
        2    0.000    0.000    0.000    0.000 _dtype.py:34(__str__)
        1    0.000    0.000    0.000    0.000 _weakrefset.py:81(add)
    67272    0.023    0.000    0.038    0.000 config.py:158(x64_enabled)
    15334    0.008    0.000    0.019    0.000 config.py:181(__getattr__)
    15334    0.008    0.000    0.012    0.000 config.py:79(read)
    15334    0.004    0.000    0.004    0.000 config.py:92(check_exists)
        7    0.000    0.000    0.000    0.000 converter.py:138(absl_to_standard)
    10970    0.014    0.000    0.073    0.000 core.py:1036(__init__)
    10506    0.006    0.000    0.006    0.000 core.py:1060(__eq__)
    13095    0.012    0.000    0.017    0.000 core.py:1066(__hash__)
     3642    0.010    0.000    0.049    0.000 core.py:1115(__init__)
     7301    0.009    0.000    0.060    0.000 core.py:1185(raise_to_shaped)
     7301    0.005    0.000    0.048    0.000 core.py:1198(<lambda>)
    10970    0.006    0.000    0.024    0.000 core.py:1211(canonicalize_shape)
    10970    0.007    0.000    0.014    0.000 core.py:1232(canonicalize_named_shape)
       17    0.000    0.000    0.001    0.000 core.py:1717(pp_eqn_compact)
       17    0.000    0.000    0.000    0.000 core.py:1718(<dictcomp>)
        7    0.000    0.000    0.000    0.000 core.py:1776(pp_kv_pair)
       17    0.000    0.000    0.000    0.000 core.py:1783(pp_kv_pairs)
        6    0.000    0.000    0.000    0.000 core.py:1785(<listcomp>)
     7657    0.013    0.000    0.750    0.000 core.py:279(bind)
     7657    0.004    0.000    0.004    0.000 core.py:383(__init__)
    13071    0.006    0.000    0.009    0.000 core.py:388(full_raise)
    13071    0.001    0.000    0.001    0.000 core.py:618(pure)
     7657    0.007    0.000    0.656    0.000 core.py:621(process_primitive)
     7657    0.007    0.000    0.014    0.000 core.py:659(with_cur_sublevel)
     7657    0.003    0.000    0.003    0.000 core.py:735(cur_sublevel)
     7657    0.002    0.000    0.005    0.000 core.py:808(full_lower)
     7657    0.018    0.000    0.048    0.000 core.py:814(find_top_trace)
     7657    0.003    0.000    0.006    0.000 core.py:815(<genexpr>)
     3642    0.006    0.000    0.057    0.000 core.py:907(concrete_aval)
     3642    0.002    0.000    0.059    0.000 core.py:917(get_aval)
     1337    0.001    0.000    0.001    0.000 core.py:954(concrete_or_error)
    10970    0.008    0.000    0.022    0.000 core.py:970(__init__)
        1    0.001    0.001    1.538    1.538 dftd3-profile.py:6(main)
        1    0.005    0.005    1.537    1.537 dftd3.py:74(d3)
        1    0.000    0.000    0.000    0.000 dftd3.py:97(<listcomp>)
    13168    0.003    0.000    0.006    0.000 dtypes.py:167(_issubclass)
    13168    0.013    0.000    0.049    0.000 dtypes.py:178(issubdtype)
    27075    0.043    0.000    0.155    0.000 dtypes.py:227(_jax_type)
   108300    0.012    0.000    0.012    0.000 dtypes.py:230(<genexpr>)
        4    0.000    0.000    0.000    0.000 dtypes.py:267(_least_upper_bound)
       11    0.000    0.000    0.000    0.000 dtypes.py:288(<genexpr>)
    38719    0.019    0.000    0.019    0.000 dtypes.py:311(is_weakly_typed)
     3597    0.003    0.000    0.003    0.000 dtypes.py:317(is_python_scalar)
    38674    0.021    0.000    0.082    0.000 dtypes.py:323(dtype)
    21661    0.018    0.000    0.181    0.000 dtypes.py:328(_result_type_raw)
     5414    0.006    0.000    0.071    0.000 dtypes.py:331(<setcomp>)
    15431    0.013    0.000    0.127    0.000 dtypes.py:333(result_type)
        6    0.000    0.000    0.000    0.000 dtypes.py:73(canonicalize_dtype)
    14470    0.002    0.000    0.002    0.000 fromnumeric.py:1922(_shape_dispatcher)
    14470    0.005    0.000    0.007    0.000 fromnumeric.py:1926(shape)
        1    0.000    0.000    0.000    0.000 genericpath.py:121(_splitext)
        8    0.000    0.000    0.000    0.000 lax.py:101(_identity)
       45    0.000    0.000    0.008    0.000 lax.py:109(sign)
     3642    0.012    0.000    0.237    0.000 lax.py:1478(_device_put_raw)
      591    0.001    0.000    0.043    0.000 lax.py:188(exp)
        6    0.000    0.000    0.000    0.000 lax.py:1967(<lambda>)
        1    0.000    0.000    0.000    0.000 lax.py:1968(<lambda>)
        1    0.000    0.000    0.000    0.000 lax.py:1969(<lambda>)
       17    0.000    0.000    0.001    0.000 lax.py:1984(standard_abstract_eval)
       41    0.000    0.000    0.000    0.000 lax.py:1985(<genexpr>)
       10    0.000    0.000    0.002    0.000 lax.py:2025(standard_translate)
       20    0.000    0.000    0.000    0.000 lax.py:2026(<genexpr>)
        9    0.000    0.000    0.000    0.000 lax.py:2030(unop_dtype_rule)
       27    0.000    0.000    0.000    0.000 lax.py:2031(<genexpr>)
        9    0.000    0.000    0.000    0.000 lax.py:2048(<lambda>)
        7    0.000    0.000    0.000    0.000 lax.py:2051(naryop_dtype_rule)
        7    0.000    0.000    0.000    0.000 lax.py:2052(<listcomp>)
       42    0.000    0.000    0.000    0.000 lax.py:2054(<genexpr>)
        7    0.000    0.000    0.000    0.000 lax.py:2074(_broadcasting_shape_rule)
        7    0.000    0.000    0.000    0.000 lax.py:2075(<listcomp>)
        5    0.000    0.000    0.000    0.000 lax.py:2087(_standard_weak_type_rule)
       10    0.000    0.000    0.000    0.000 lax.py:2088(<genexpr>)
       11    0.000    0.000    0.000    0.000 lax.py:2090(_naryop_weak_type_rule)
       29    0.000    0.000    0.000    0.000 lax.py:2091(<genexpr>)
       27    0.000    0.000    0.000    0.000 lax.py:2101(<genexpr>)
        1    0.000    0.000    0.000    0.000 lax.py:2179(_sign_translation_rule)
        5    0.000    0.000    0.000    0.000 lax.py:2471(_integer_pow_dtype_rule)
        5    0.000    0.000    0.000    0.000 lax.py:2478(_integer_pow_translation_rule)
        1    0.000    0.000    0.000    0.000 lax.py:2644(_convert_element_type_shape_rule)
        1    0.000    0.000    0.000    0.000 lax.py:2647(_convert_element_type_dtype_rule)
        1    0.000    0.000    0.000    0.000 lax.py:2650(_convert_element_type_weak_type_rule)
        1    0.000    0.000    0.000    0.000 lax.py:2653(_convert_element_type_translation_rule)
       45    0.000    0.000    0.007    0.000 lax.py:299(abs)
       45    0.000    0.000    0.008    0.000 lax.py:303(pow)
     1292    0.002    0.000    0.104    0.000 lax.py:307(integer_pow)
      180    0.000    0.000    0.110    0.001 lax.py:311(sqrt)
     1865    0.002    0.000    0.147    0.000 lax.py:339(add)
     1092    0.001    0.000    0.095    0.000 lax.py:343(sub)
     1797    0.002    0.000    0.151    0.000 lax.py:347(mul)
      560    0.001    0.000    0.066    0.000 lax.py:351(div)
       55    0.000    0.000    0.011    0.000 lax.py:397(gt)
    11644    0.036    0.000    0.464    0.000 lax.py:409(convert_element_type)
        7    0.000    0.000    0.000    0.000 lax.py:6142(_check_same_dtypes)
        7    0.000    0.000    0.000    0.000 lax.py:6152(<setcomp>)
    11644    0.009    0.000    0.012    0.000 lax_numpy.py:111(__instancecheck__)
     6185    0.007    0.000    0.036    0.000 lax_numpy.py:239(_promote_shapes)
     5414    0.005    0.000    0.027    0.000 lax_numpy.py:244(<listcomp>)
     5414    0.001    0.000    0.001    0.000 lax_numpy.py:245(<listcomp>)
     4854    0.012    0.000    0.472    0.000 lax_numpy.py:269(_promote_dtypes)
     4854    0.007    0.000    0.380    0.000 lax_numpy.py:278(<listcomp>)
     1376    0.004    0.000    0.121    0.000 lax_numpy.py:280(_promote_dtypes_inexact)
     1376    0.001    0.000    0.092    0.000 lax_numpy.py:289(<listcomp>)
     1376    0.001    0.000    0.007    0.000 lax_numpy.py:291(_to_inexact_dtype)
    11644    0.003    0.000    0.026    0.000 lax_numpy.py:305(_arraylike)
     6230    0.006    0.000    0.042    0.000 lax_numpy.py:308(_check_arraylike)
    17874    0.005    0.000    0.031    0.000 lax_numpy.py:311(<genexpr>)
     6185    0.003    0.000    0.044    0.000 lax_numpy.py:317(_check_no_float0s)
    17784    0.006    0.000    0.037    0.000 lax_numpy.py:319(<genexpr>)
     4854    0.013    0.000    0.587    0.000 lax_numpy.py:329(_promote_args)
     1331    0.004    0.000    0.143    0.000 lax_numpy.py:335(_promote_args_inexact)
       45    0.000    0.000    0.001    0.000 lax_numpy.py:343(_constant_like)
     1476    0.001    0.000    0.007    0.000 lax_numpy.py:356(issubdtype)
     3597    0.003    0.000    0.007    0.000 lax_numpy.py:360(isscalar)
      771    0.001    0.000    0.225    0.000 lax_numpy.py:374(<lambda>)
     1092    0.003    0.000    0.254    0.000 lax_numpy.py:387(<lambda>)
     3662    0.009    0.000    0.718    0.000 lax_numpy.py:395(fn)
       55    0.000    0.000    0.019    0.000 lax_numpy.py:457(fn)
     6371    0.012    0.000    1.214    0.000 lax_numpy.py:5253(deferring_binary_op)
     1362    0.001    0.000    0.331    0.000 lax_numpy.py:5260(<lambda>)
      560    0.001    0.000    0.138    0.000 lax_numpy.py:536(true_divide)
     1337    0.002    0.000    0.129    0.000 lax_numpy.py:588(power)
       45    0.000    0.000    0.054    0.001 lax_numpy.py:864(cbrt)
    13116    0.002    0.000    0.002    0.000 lazy.py:80(is_trivial)
    31525    0.003    0.000    0.003    0.000 multiarray.py:644(result_type)
    26340    0.011    0.000    0.017    0.000 numerictypes.py:285(issubclass_)
    13170    0.012    0.000    0.030    0.000 numerictypes.py:359(issubdtype)
        3    0.000    0.000    0.000    0.000 os.py:670(__getitem__)
        3    0.000    0.000    0.000    0.000 os.py:748(encode)
        3    0.000    0.000    0.000    0.000 os.py:766(getenv)
        1    0.000    0.000    0.000    0.000 posixpath.py:117(splitext)
        1    0.000    0.000    0.000    0.000 posixpath.py:140(basename)
        1    0.000    0.000    0.000    0.000 posixpath.py:41(_get_sep)
      117    0.000    0.000    0.000    0.000 pprint_util.py:25(__init__)
       25    0.000    0.000    0.000    0.000 pprint_util.py:28(indent)
       25    0.000    0.000    0.000    0.000 pprint_util.py:29(<listcomp>)
        7    0.000    0.000    0.000    0.000 pprint_util.py:36(__add__)
       36    0.000    0.000    0.000    0.000 pprint_util.py:39(__rshift__)
       17    0.000    0.000    0.000    0.000 pprint_util.py:52(__str__)
       35    0.000    0.000    0.000    0.000 pprint_util.py:53(<genexpr>)
       60    0.000    0.000    0.000    0.000 pprint_util.py:56(pp)
       60    0.000    0.000    0.000    0.000 pprint_util.py:57(<listcomp>)
        6    0.000    0.000    0.000    0.000 pprint_util.py:62(vcat)
        1    0.000    0.000    0.000    0.000 threading.py:1031(name)
        1    0.000    0.000    0.000    0.000 threading.py:1306(current_thread)
        1    0.000    0.000    0.000    0.000 threading.py:81(RLock)
    34/17    0.001    0.000    0.197    0.012 util.py:189(cached)
7674/7657    0.019    0.000    0.244    0.000 util.py:193(wrapper)
        6    0.000    0.000    0.000    0.000 util.py:206(memoized)
    59592    0.037    0.000    0.069    0.000 util.py:210(wrapper)
       17    0.000    0.000    0.000    0.000 util.py:30(safe_zip)
    18627    0.034    0.000    0.045    0.000 util.py:36(safe_map)
       17    0.000    0.000    0.000    0.000 util.py:43(unzip2)
       55    0.000    0.000    0.000    0.000 util.py:99(__get__)
       45    0.000    0.000    0.000    0.000 utils.py:120(lin)
       55    0.012    0.000    0.861    0.016 utils.py:130(getc6)
        1    0.000    0.000    0.000    0.000 utils.py:172(check_inputs)
        1    0.002    0.002    0.321    0.321 utils.py:78(ncoord)
        1    0.000    0.000    0.000    0.000 utils.py:95(<listcomp>)
       55    0.000    0.000    0.001    0.000 xla.py:1019(_forward_method)
    11299    0.008    0.000    0.023    0.000 xla.py:1033(make_device_array)
    24360    0.028    0.000    0.028    0.000 xla.py:1075(__init__)
       17    0.000    0.000    0.000    0.000 xla.py:108(aval_to_result_handler)
       55    0.000    0.000    0.000    0.000 xla.py:1101(_check_if_deleted)
       55    0.000    0.000    0.000    0.000 xla.py:1118(_value)
    10828    0.002    0.000    0.002    0.000 xla.py:1126(shape)
    27768    0.005    0.000    0.005    0.000 xla.py:1130(dtype)
     3659    0.004    0.000    0.032    0.000 xla.py:114(array_result_handler)
    16713    0.016    0.000    0.191    0.000 xla.py:126(device_put)
    13061    0.007    0.000    0.073    0.000 xla.py:1296(_device_put_device_array)
    13061    0.032    0.000    0.059    0.000 xla.py:1302(_copy_device_array_to_device)
    13116    0.005    0.000    0.007    0.000 xla.py:1323(_force)
     3652    0.023    0.000    0.047    0.000 xla.py:133(_device_put_array)
    16713    0.010    0.000    0.055    0.000 xla.py:149(canonicalize_dtype)
     3642    0.005    0.000    0.042    0.000 xla.py:160(_canonicalize_ndarray_dtype)
       10    0.000    0.000    0.000    0.000 xla.py:163(_canonicalize_python_scalar_dtype)
    13071    0.006    0.000    0.008    0.000 xla.py:173(abstractify)
       10    0.000    0.000    0.000    0.000 xla.py:184(_make_abstract_python_scalar)
        7    0.000    0.000    0.000    0.000 xla.py:210(_param_uses_outfeed)
       17    0.000    0.000    0.000    0.000 xla.py:219(primitive_uses_outfeed)
    13071    0.005    0.000    0.013    0.000 xla.py:232(arg_spec)
     7657    0.019    0.000    0.649    0.000 xla.py:239(apply_primitive)
       17    0.000    0.000    0.196    0.012 xla.py:253(xla_primitive_callable)
       17    0.000    0.000    0.000    0.000 xla.py:297(_device_from_arg_devices)
       17    0.000    0.000    0.000    0.000 xla.py:308(<setcomp>)
       17    0.001    0.000    0.008    0.000 xla.py:314(primitive_computation)
       17    0.185    0.011    0.185    0.011 xla.py:352(backend_compile)
     7657    0.213    0.000    0.373    0.000 xla.py:357(_execute_compiled_primitive)
    20728    0.007    0.000    0.103    0.000 xla.py:359(<genexpr>)
     7657    0.020    0.000    0.039    0.000 xla.py:375(check_special)
    13061    0.001    0.000    0.001    0.000 xla.py:75(identity)
       17    0.000    0.000    0.002    0.000 xla.py:788(_xla_callable_args)
       17    0.000    0.000    0.002    0.000 xla.py:806(<listcomp>)
       24    0.000    0.000    0.001    0.000 xla.py:834(_xla_param)
       24    0.000    0.000    0.000    0.000 xla.py:86(_make_array_shape)
       24    0.000    0.000    0.000    0.000 xla.py:96(aval_to_xla_shapes)
        2    0.000    0.000    0.023    0.012 xla_bridge.py:126(_get_local_backend)
        3    0.000    0.000    0.023    0.008 xla_bridge.py:166(get_backend)
    16730    0.014    0.000    0.038    0.000 xla_bridge.py:181(get_device_backend)
       17    0.000    0.000    0.000    0.000 xla_bridge.py:187(device_count)
       24    0.000    0.000    0.001    0.000 xla_bridge.py:316(parameter)
       17    0.003    0.000    0.003    0.000 xla_bridge.py:407(make_computation_builder)
       17    0.000    0.000    0.000    0.000 xla_bridge.py:66(get_compile_options)
        2    0.000    0.000    0.023    0.011 xla_client.py:116(_get_local_backends)
        2    0.000    0.000    0.023    0.011 xla_client.py:140(get_local_backend)
       17    0.000    0.000    0.000    0.000 xla_client.py:166(__init__)
        1    0.000    0.000    0.000    0.000 xla_client.py:216(dtype_to_etype)
        1    0.000    0.000    0.003    0.003 xla_client.py:61(_interpreter_backend_factory)
        1    0.000    0.000    0.017    0.017 xla_client.py:65(_cpu_backend_factory)
        1    0.000    0.000    0.002    0.002 xla_client.py:69(_gpu_backend_factory)
        1    0.000    0.000    0.000    0.000 xla_client.py:96(_tpu_backend_factory)
       17    0.000    0.000    0.000    0.000 {built-in method __new__ of type object at 0x7f8a9601b5c0}
     1337    0.000    0.000    0.000    0.000 {built-in method _operator.index}
        2    0.000    0.000    0.000    0.000 {built-in method _thread.get_ident}
       33    0.000    0.000    0.000    0.000 {built-in method builtins.all}
    39524    0.025    0.000    0.106    0.000 {built-in method builtins.any}
        1    0.000    0.000    1.538    1.538 {built-in method builtins.exec}
     3733    0.001    0.000    0.001    0.000 {built-in method builtins.getattr}
    15243    0.002    0.000    0.002    0.000 {built-in method builtins.hasattr}
    13095    0.003    0.000    0.003    0.000 {built-in method builtins.hash}
91008/79364    0.023    0.000    0.032    0.000 {built-in method builtins.isinstance}
    52682    0.009    0.000    0.009    0.000 {built-in method builtins.issubclass}
    66918    0.007    0.000    0.007    0.000 {built-in method builtins.len}
     7719    0.010    0.000    0.016    0.000 {built-in method builtins.max}
       45    0.000    0.000    0.000    0.000 {built-in method builtins.min}
       24    0.000    0.000    0.000    0.000 {built-in method builtins.next}
    10987    0.005    0.000    0.005    0.000 {built-in method builtins.sorted}
        6    0.000    0.000    0.000    0.000 {built-in method builtins.sum}
     7657    0.001    0.000    0.001    0.000 {built-in method from_iterable}
       24    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.array_shape}
        1    0.017    0.017    0.017    0.017 {built-in method jaxlib.xla_extension.get_cpu_client}
        1    0.002    0.002    0.002    0.002 {built-in method jaxlib.xla_extension.get_gpu_client}
        1    0.003    0.003    0.003    0.003 {built-in method jaxlib.xla_extension.get_interpreter_client}
        1    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.get_tpu_client}
    67272    0.015    0.000    0.015    0.000 {built-in method jaxlib.xla_extension.jax_jit.get_enable_x64}
        1    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.ops.Abs}
        1    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.ops.Add}
        1    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.ops.ConvertElementType}
        1    0.001    0.001    0.001    0.001 {built-in method jaxlib.xla_extension.ops.Div}
        1    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.ops.Exp}
        1    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.ops.Gt}
       18    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.ops.Mul}
       24    0.001    0.000    0.001    0.000 {built-in method jaxlib.xla_extension.ops.Parameter}
        1    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.ops.Pow}
        1    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.ops.Sign}
        1    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.ops.Sqrt}
        1    0.000    0.000    0.000    0.000 {built-in method jaxlib.xla_extension.ops.Sub}
     7339    0.011    0.000    0.011    0.000 {built-in method numpy.array}
    45995    0.052    0.000    0.065    0.000 {built-in method numpy.core._multiarray_umath.implement_array_function}
        1    0.000    0.000    0.000    0.000 {built-in method numpy.zeros}
        2    0.000    0.000    0.000    0.000 {built-in method posix.fspath}
        1    0.000    0.000    0.000    0.000 {built-in method posix.getpid}
        1    0.000    0.000    0.000    0.000 {built-in method sys._getframe}
        1    0.000    0.000    0.000    0.000 {built-in method time.time}
        8    0.000    0.000    0.000    0.000 {method 'acquire' of '_thread.RLock' objects}
        1    0.000    0.000    0.000    0.000 {method 'add' of 'set' objects}
      154    0.000    0.000    0.000    0.000 {method 'append' of 'list' objects}
       10    0.000    0.000    0.000    0.000 {method 'capitalize' of 'str' objects}
       90    0.000    0.000    0.000    0.000 {method 'casefold' of 'str' objects}
        1    0.000    0.000    0.000    0.000 {method 'disable' of '_lsprof.Profiler' objects}
        3    0.000    0.000    0.000    0.000 {method 'encode' of 'str' objects}
        1    0.000    0.000    0.000    0.000 {method 'find' of 'str' objects}
        1    0.000    0.000    0.000    0.000 {method 'flush' of '_io.TextIOWrapper' objects}
        2    0.000    0.000    0.000    0.000 {method 'format' of 'str' objects}
    44373    0.005    0.000    0.005    0.000 {method 'get' of 'dict' objects}
        4    0.000    0.000    0.000    0.000 {method 'intersection' of 'set' objects}
    23208    0.018    0.000    0.018    0.000 {method 'item' of 'numpy.generic' objects}
        1    0.000    0.000    0.000    0.000 {method 'items' of 'collections.OrderedDict' objects}
    24099    0.003    0.000    0.003    0.000 {method 'items' of 'dict' objects}
       27    0.000    0.000    0.000    0.000 {method 'join' of 'str' objects}
        1    0.000    0.000    0.000    0.000 {method 'lower' of 'str' objects}
    10943    0.003    0.000    0.003    0.000 {method 'mro' of 'type' objects}
        9    0.000    0.000    0.000    0.000 {method 'pop' of 'dict' objects}
        4    0.000    0.000    0.000    0.000 {method 'pop' of 'set' objects}
        8    0.000    0.000    0.000    0.000 {method 'release' of '_thread.RLock' objects}
        3    0.000    0.000    0.000    0.000 {method 'rfind' of 'str' objects}
        1    0.000    0.000    0.000    0.000 {method 'search' of 're.Pattern' objects}
       10    0.000    0.000    0.000    0.000 {method 'split' of 'str' objects}
       60    0.000    0.000    0.000    0.000 {method 'splitlines' of 'str' objects}
        1    0.000    0.000    0.000    0.000 {method 'values' of 'collections.OrderedDict' objects}
       17    0.000    0.000    0.000    0.000 {method 'values' of 'dict' objects}
        1    0.000    0.000    0.000    0.000 {method 'write' of '_io.TextIOWrapper' objects}

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment