diff options
Diffstat (limited to 'unsupported/test/cxx11_tensor_contract_sycl.cpp')
| -rw-r--r-- | unsupported/test/cxx11_tensor_contract_sycl.cpp | 36 |
1 files changed, 18 insertions, 18 deletions
diff --git a/unsupported/test/cxx11_tensor_contract_sycl.cpp b/unsupported/test/cxx11_tensor_contract_sycl.cpp index 41acd5579..5bace66c5 100644 --- a/unsupported/test/cxx11_tensor_contract_sycl.cpp +++ b/unsupported/test/cxx11_tensor_contract_sycl.cpp @@ -254,26 +254,26 @@ void test_sycl_contraction_sizes(const Device& sycl_device) { template <typename Dev_selector> void tensorContractionPerDevice(Dev_selector& s){ QueueInterface queueInterface(s); auto sycl_device=Eigen::SyclDevice(&queueInterface); - test_sycl_contraction<ColMajor, float,ptrdiff_t>(sycl_device, 32, 32, 32); - test_sycl_contraction<RowMajor,float,ptrdiff_t>(sycl_device, 32, 32, 32); - test_scalar<ColMajor,float,ptrdiff_t>(sycl_device, 32, 32, 32); - test_scalar<RowMajor,float,ptrdiff_t>(sycl_device, 32, 32, 32); + test_sycl_contraction<ColMajor, float,int64_t>(sycl_device, 32, 32, 32); + test_sycl_contraction<RowMajor,float,int64_t>(sycl_device, 32, 32, 32); + test_scalar<ColMajor,float,int64_t>(sycl_device, 32, 32, 32); + test_scalar<RowMajor,float,int64_t>(sycl_device, 32, 32, 32); std::chrono::time_point<std::chrono::system_clock> start, end; start = std::chrono::system_clock::now(); - test_sycl_contraction<ColMajor,float,ptrdiff_t>(sycl_device, 128, 128, 128); - test_sycl_contraction<RowMajor,float,ptrdiff_t>(sycl_device, 128, 128, 128); - test_scalar<ColMajor,float,ptrdiff_t>(sycl_device, 128, 128, 128); - test_scalar<RowMajor,float,ptrdiff_t>(sycl_device, 128, 128, 128); - test_sycl_contraction_m<ColMajor, float, ptrdiff_t>(sycl_device); - test_sycl_contraction_m<RowMajor, float, ptrdiff_t>(sycl_device); - test_sycl_contraction_n<ColMajor, float, ptrdiff_t>(sycl_device); - test_sycl_contraction_n<RowMajor, float, ptrdiff_t>(sycl_device); - test_sycl_contraction_k<ColMajor, float, ptrdiff_t>(sycl_device); - test_sycl_contraction_k<RowMajor, float, ptrdiff_t>(sycl_device); - test_sycl_contraction_sizes<ColMajor, float, ptrdiff_t>(sycl_device); - test_sycl_contraction_sizes<RowMajor, float, ptrdiff_t>(sycl_device); - test_TF<RowMajor, float, ptrdiff_t>(sycl_device); - test_TF<ColMajor, float, ptrdiff_t>(sycl_device); + test_sycl_contraction<ColMajor,float,int64_t>(sycl_device, 128, 128, 128); + test_sycl_contraction<RowMajor,float,int64_t>(sycl_device, 128, 128, 128); + test_scalar<ColMajor,float,int64_t>(sycl_device, 128, 128, 128); + test_scalar<RowMajor,float,int64_t>(sycl_device, 128, 128, 128); + test_sycl_contraction_m<ColMajor, float, int64_t>(sycl_device); + test_sycl_contraction_m<RowMajor, float, int64_t>(sycl_device); + test_sycl_contraction_n<ColMajor, float, int64_t>(sycl_device); + test_sycl_contraction_n<RowMajor, float, int64_t>(sycl_device); + test_sycl_contraction_k<ColMajor, float, int64_t>(sycl_device); + test_sycl_contraction_k<RowMajor, float, int64_t>(sycl_device); + test_sycl_contraction_sizes<ColMajor, float, int64_t>(sycl_device); + test_sycl_contraction_sizes<RowMajor, float, int64_t>(sycl_device); + test_TF<RowMajor, float, int64_t>(sycl_device); + test_TF<ColMajor, float, int64_t>(sycl_device); end = std::chrono::system_clock::now(); std::chrono::duration<double> elapsed_seconds = end-start; |