Fixing TensorArgMaxSycl.h; Removing warning related to the hardcoded type of dims to be int in Argmax.

1 files changed, 2 insertions, 7 deletions

diff --git a/unsupported/test/cxx11_tensor_argmax_sycl.cpp b/unsupported/test/cxx11_tensor_argmax_sycl.cpp
index b80cb8f06..521a7f82c 100644
--- a/unsupported/test/cxx11_tensor_argmax_sycl.cpp
+++ b/unsupported/test/cxx11_tensor_argmax_sycl.cpp
@@ -143,10 +143,6 @@ static void test_sycl_argmax_dim(const Eigen::SyclDevice &sycl_device)
   }
 }
 
-
-
-
-
 template <typename DataType, int DataLayout, typename DenseIndex>
 static void test_sycl_argmin_dim(const Eigen::SyclDevice &sycl_device)
 {
@@ -242,9 +238,8 @@ template<typename DataType, typename Device_Selector> void sycl_argmax_test_per_
 }
 
 void test_cxx11_tensor_argmax_sycl() {
-// TODO {lukei}: re-enable once fixed
-/* for (const auto& device :Eigen::get_sycl_supported_devices()) {
+ for (const auto& device :Eigen::get_sycl_supported_devices()) {
     CALL_SUBTEST(sycl_argmax_test_per_device<double>(device));
   }
-*/
+
 }