Skip to content

Fix handling of JAX tensor types without TVM FFI and leading dim deduction for all-1-size tensors - #3583

Open
dextero wants to merge 2 commits into
NVIDIA:mainfrom
dextero:main
Open

Fix handling of JAX tensor types without TVM FFI and leading dim deduction for all-1-size tensors#3583
dextero wants to merge 2 commits into
NVIDIA:mainfrom
dextero:main

[CuTeDSL] Register lazy JIT argument adapter for JAX arrays

3a59277
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs