Eric Shi

@gdaviet Your fix seems to have worked, I haven't seen this error since then.

@MiguelZamoraM added an example to do this in #419.

@steinraf Sorry for the extremely late reply. I examined your reproducer and believe the issue comes down to the `wp.HashGrid` from the first iteration being garbage-collected before the `hashgrid_tester` kernel...

Update: I was missing these warnings: ``` test_dlpack_warp_to_jax_cpu (__main__.TestDLPack) ... /home/eshi/code-projects/warp/warp/tests/interop/test_dlpack.py:388: DeprecationWarning: Calling from_dlpack with a DLPack tensor is deprecated. The argument to from_dlpack should be an array from another...

https://github.com/jax-ml/jax/discussions/6055 shows us how to create a NumPy array with an alignment that JAX is happy with.

This comment indicates that XLA needs 16-byte alignment: https://github.com/openxla/xla/blob/b5c0101d998781261b2815ef03146e9ff28cbb32/xla/pjrt/cpu/abstract_tfrt_cpu_buffer.cc#L771-L776 But this value can change depending on the value of `EIGEN_MAX_ALIGN_BYTES`: https://github.com/openxla/xla/blob/main/xla/backends/cpu/alignment.h#L32

@nvlukasz suggests adding a config option to control alignment, e.g. `wp.config.cpu_align = 16`

I added dedaebce756d6f9380c182e78e080f4b74e3ddfd to temporarily avoid the test failure for now.

@AnkaChan added a `rebuild()` method as requested. See 3ece71f796f5ab72c4c8b4c6d40fc17b5b5b0c9c. It will be in Warp v1.10 (scheduled for beginning of November) but you can start using it sooner by building Warp...

@WenchaoHuang: @StafaH's recent commit 0e08171333f80307c4ed1d2384a8b09c3d54dc42 adds a feature that should address your rayTime request.