(host_ptr, num_floats)
| 73 | |
| 74 | |
| 75 | def _float_view(host_ptr, num_floats): |
| 76 | return np.frombuffer( |
| 77 | (ctypes.c_float * num_floats).from_address(host_ptr), |
| 78 | dtype=np.float32, |
| 79 | count=num_floats, |
| 80 | ) |
| 81 | |
| 82 | |
| 83 | # CUDA C++: vector add with grid-stride loop |