Skip to content

Commit 18c71df

Browse files
committed
Added runtime errors for host.
Signed-off-by: JackAKirk <[email protected]>
1 parent 48177cc commit 18c71df

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

sycl/include/sycl/ext/oneapi/matrix/matrix-tensorcore.hpp

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,23 @@ template <typename type, size_t size> class wi_data {
2222

2323
public:
2424
wi_data(marray<type, size> &wi_data) : data(wi_data){};
25-
size_t length() { return data.size(); };
25+
size_t length() {
26+
#if defined(__SYCL_DEVICE_ONLY__) && defined(__NVPTX__)
27+
return data.size();
28+
#else
29+
throw runtime_error("joint matrix is not supported on host device.",
30+
PI_INVALID_DEVICE);
31+
#endif // defined(__SYCL_DEVICE_ONLY__) && defined(__NVPTX__)
32+
};
2633

27-
type &operator[](size_t i) { return data[i]; };
34+
type &operator[](size_t i) {
35+
#if defined(__SYCL_DEVICE_ONLY__) && defined(__NVPTX__)
36+
return data[i];
37+
#else
38+
throw runtime_error("joint matrix is not supported on host device.",
39+
PI_INVALID_DEVICE);
40+
#endif // defined(__SYCL_DEVICE_ONLY__) && defined(__NVPTX__)
41+
};
2842
};
2943

3044
template <typename T, matrix_use Use, size_t Rows = sycl::dynamic_extent,

0 commit comments

Comments
 (0)