You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
But on GPU, the data is not copied during prepare_inputs, only when the data is accessed during run_benchmark. The timings I get for Bohrium are thus higher than for other libraries, which copy the data right away.
I guess you can call bohrium._bh.get_data_pointer(bh_ary, copy2host=False, allocate=True), which will return a point (as a Python integter) to the GPU memory. If the data is on main memory, it will be copied to the GPU before returning.
Activity
madsbk commentedon Nov 7, 2019
I guess you can call
bohrium._bh.get_data_pointer(bh_ary, copy2host=False, allocate=True), which will return a point (as a Python integter) to the GPU memory. If the data is on main memory, it will be copied to the GPU before returning.