changed linalg_traits definiton