gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[dnn/gemm] added some bounds checking

Browse Source
This commit is contained in:
Philippe Tillet
2019-07-19 21:32:55 -07:00
parent 5215fb0424
commit 28c250216c
5 changed files with 36 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/cpp/dot.cpp
View File

@@ -8,15 +8,15 @@
int main() {
bool AT = true;
bool BT = false;
bool AT = false;
bool BT = true;
typedef float T;
std::string ty = "fp16";
size_t dt_nbytes = sizeof(T);
// initialize default compute device
auto context = triton::driver::backend::contexts::get_default();
// matrix multiplication parameters
int32_t M = 65536, N = 2048, K = 2048;
int32_t M = 4096, N = 4096, K = 4096;
std::vector<T> hc(M*N);
std::vector<T> rc(M*N);
std::vector<T> ha(M*K);