1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
# libc++ issue: https://github.com/llvm/llvm-project/issues/98734
diff --git a/library/src/amd_detail/rocblaslt/src/rocblaslt_mat.cpp b/library/src/amd_detail/rocblaslt/src/rocblaslt_mat.cpp
index 82cc81f..94e62d4 100644
--- a/library/src/amd_detail/rocblaslt/src/rocblaslt_mat.cpp
+++ b/library/src/amd_detail/rocblaslt/src/rocblaslt_mat.cpp
@@ -28,6 +28,7 @@
#include "handle.h"
#include "rocblaslt_mat_utils.hpp"
#include "tensile_host.hpp"
+#include <array>
#include <hip/hip_runtime_api.h>
@@ -602,7 +603,7 @@ rocblaslt_status
std::vector<int64_t> ldc_vec, batch_stride_c_vec, num_batches_c_vec;
std::vector<int64_t> ldd_vec, batch_stride_d_vec, num_batches_d_vec;
std::vector<int64_t> lde_vec, batch_stride_e_vec, num_batches_e_vec;
- std::vector<int8_t[16]> alpha_1(matmul_descr.size());
+ std::vector<std::array<int8_t, 16>> alpha_1(matmul_descr.size());
std::vector<bool> gradient_vec;
@@ -692,10 +693,10 @@ rocblaslt_status
return validArgs;
const void* alphaTmp = nullptr;
- memset(alpha_1[i], 0, sizeof(int8_t) * 16);
+ memset(alpha_1[i].data(), 0, sizeof(int8_t) * 16);
if(scaleAlphaVec)
{
- setTo1(compute_type, (void*)alpha_1[i], &alphaTmp);
+ setTo1(compute_type, (void*)alpha_1[i].data(), &alphaTmp);
}
else
{
@@ -867,7 +868,7 @@ rocblaslt_status
std::vector<int64_t> lde_vec, batch_stride_e_vec, num_batches_e_vec;
std::vector<bool> gradient_vec;
- std::vector<int8_t[16]> alpha_1(m.size());
+ std::vector<std::array<int8_t, 16>> alpha_1(m.size());
for(int i = 0; i < m.size(); i++)
{
@@ -924,10 +925,10 @@ rocblaslt_status
return validArgs;
const void* alphaTmp = nullptr;
- memset(alpha_1[i], 0, sizeof(int8_t) * 16);
+ memset(alpha_1[i].data(), 0, sizeof(int8_t) * 16);
if(scaleAlphaVec)
{
- setTo1(compute_type, (void*)alpha_1[i], &alphaTmp);
+ setTo1(compute_type, (void*)alpha_1[i].data(), &alphaTmp);
}
else
{
|
GitWeb
