[bsr_add] fix the cuda bsr add path and add the cpu one#19
[bsr_add] fix the cuda bsr add path and add the cpu one#19zitongzhan merged 3 commits intoreleasefrom
Conversation
|
@copilot This PR makes it possible for CPU to perform the BSR add operation. Therefore, many test verifying the correctness of Jacobian that was skipped on non-CUDA device can be relaxed and added back. Identify them. |
… add is supported Agent-Logs-Url: https://github.com/sair-lab/bae/sessions/c8ccdd05-bc87-417a-80b2-a438ff812653 Co-authored-by: zitongzhan <24406547+zitongzhan@users.noreply.github.com>
Identified and addressed in commit
|
…PU runners Agent-Logs-Url: https://github.com/sair-lab/bae/sessions/aff0ed51-66f5-4040-be7d-8c19786ebaee Co-authored-by: zitongzhan <24406547+zitongzhan@users.noreply.github.com>
tests/autograd/test_bal_jacobian.pyby restoring the CUDA-only guardstests/sparse/test_bsr.pyback to CUDA-only