4 ms·

Same for pytorch. I don't know enough pyt, but guessing they didn't jit anything. https://github.com/utkarsh530/GPUODEBenchmarks/blob/ef807198584e4634c82bfd6a

by yablak 3y ago

Same for pytorch. I don't know enough pyt, but guessing they didn't jit anything. https://github.com/utkarsh530/GPUODEBenchmarks/blob/ef807198584e4634c82bfd6a82bb9241c51d56d1/GPU_ODE_PyTorch/bench_torchdiffeq.py#L82 https://github.com/utkarsh530/GPUODEBenchmarks/blob/ef807198...