bf5d19991c3802e90b2b424a6ec3ac07515e91ae
BI-V100 base image does not have libcusolver.so at: /opt/sw_home/local/cuda/lib64/libcusolver.so torch.linalg.solve_triangular requires cuSOLVER which is missing. Replace with row-by-row forward substitution using only basic matmul and indexing ops (torch.zeros_like, matmul, indexing). The linear_attention gated_delta_rule solves (I-A)@X=RHS where A is strictly lower-triangular. Forward sub: x[0]=rhs[0], x[i]=rhs[i]+A[i,:i]@x[:i]. Mathematically equivalent.
project_6
Description
Languages
C++
41.8%
Cuda
31.6%
Python
22.2%
C
2.1%
CMake
1.1%
Other
1.1%