Files
rocm-systems/test/ctrl/test.cpp
T

19 خطوط
571 B
C++

2017-11-09 17:26:19 -06:00
#include <hsa.h>
#include <string.h>
#include <iostream>
#include "ctrl/run_kernel.h"
#include "ctrl/test_aql.h"
#include "simple_convolution/simple_convolution.h"
int main(int argc, char** argv) {
2018-01-22 17:02:46 -06:00
const char* kiter_s = getenv("ROCP_KITER");
const char* diter_s = getenv("ROCP_DITER");
const int kiter = (kiter_s != NULL) ? atol(kiter_s) : 1;
const int diter = (diter_s != NULL) ? atol(diter_s) : 1;
2017-11-09 17:26:19 -06:00
TestHsa::HsaInstantiate();
2018-01-22 17:02:46 -06:00
for (int i = 0; i < kiter; ++i) RunKernel<SimpleConvolution, TestAql>(argc, argv, diter);
TestHsa::HsaShutdown();
2017-11-09 17:26:19 -06:00
return 0;
}