add Nan checks in linalg