Bugfix : Fix bug in all the templates except vector-axpy