#pragma once

#include "./core/omp_config.h"
#include "detail/argmax_serial.h"



namespace numerics{

	// ---------------- Vector -> Scalar ----------------
	template <typename T>
	inline uint64_t argmax(const utils::Vector<T>& v) {
		return detail::argmax_serial(v);
	}


	// ---------------- Matrix -> Scalar ----------------
	template <typename T>
	inline utils::Vector<uint64_t> argmax(const utils::Matrix<T>& A) {
		return detail::argmax_serial(A);
	}

	// ---------------- Matrix -> Vector ----------------
	template <typename T>
	inline utils::Vector<uint64_t> argmax_rowwise(const utils::Matrix<T>& A) {
		return detail::argmax_rowwise_serial(A);
	}

	template <typename T>
	inline utils::Vector<uint64_t> argmax_colwise(const utils::Matrix<T>& A) {
		return detail::argmax_colwise_serial(A);
	}
}