diff options
Diffstat (limited to 'tests/projects/ascendc/mixed_binary/src')
| -rw-r--r-- | tests/projects/ascendc/mixed_binary/src/helper.aicpu | 6 | ||||
| -rw-r--r-- | tests/projects/ascendc/mixed_binary/src/main.asc | 25 |
2 files changed, 31 insertions, 0 deletions
diff --git a/tests/projects/ascendc/mixed_binary/src/helper.aicpu b/tests/projects/ascendc/mixed_binary/src/helper.aicpu new file mode 100644 index 000000000..a11afdd90 --- /dev/null +++ b/tests/projects/ascendc/mixed_binary/src/helper.aicpu @@ -0,0 +1,6 @@ +#include "aicpu_api.h" + +__global__ __aicpu__ uint32_t hello_world(void *args) { + AscendC::printf("Hello World!!!\n"); + return 0; +} diff --git a/tests/projects/ascendc/mixed_binary/src/main.asc b/tests/projects/ascendc/mixed_binary/src/main.asc new file mode 100644 index 000000000..9bc0999bd --- /dev/null +++ b/tests/projects/ascendc/mixed_binary/src/main.asc @@ -0,0 +1,25 @@ +#include "acl/acl.h" + +struct KernelArgs { + int mode; +}; + +extern __global__ __aicpu__ uint32_t hello_world(void *args); + +int32_t main(int argc, char const *argv[]) { + aclInit(nullptr); + int32_t deviceId = 0; + aclrtSetDevice(deviceId); + aclrtStream stream = nullptr; + aclrtCreateStream(&stream); + + struct KernelArgs args = {0}; + constexpr uint32_t numBlocks = 1; + hello_world<<<numBlocks, nullptr, stream>>>(&args, sizeof(KernelArgs)); + aclrtSynchronizeStream(stream); + + aclrtDestroyStream(stream); + aclrtResetDevice(deviceId); + aclFinalize(); + return 0; +} |
