diff options
| author | wuzhenqing <[email protected]> | 2026-05-04 11:15:50 +0000 |
|---|---|---|
| committer | wuzhenqing <[email protected]> | 2026-05-04 11:18:09 +0000 |
| commit | fa807b737b4b6b3004cccafc0dfdfdd66fcdce2d (patch) | |
| tree | 558dc6bc9e5749180fa97a569aab8bb6c2e54fec /tests/modules/language/ascendc/test.lua | |
| parent | 8bbbb2860fdd1910db466fe318ee8d465728ca32 (diff) | |
feat(ascendc): add Huawei Ascend C toolchain support
Add toolchain, language, rules, and compiler modules for the
bisheng compiler driver from CANN (Compute Architecture for
Neural Networks) toolkit. Supports .asc (Ascend C kernel) and
.aicpu (AI-CPU operator) source kinds with binary/static/shared
target types.
- Toolchain: SDK auto-detection via ASCEND_HOME_PATH env or --sdk
- Compiler: bisheng module inherits from clang, adds add_sourceflags
for explicit sourcekind overrides
- Rules: ascendc.env (default C++17), npuarchs (--npu-arch), build
- Language: check_main detection, add_ascnpuarchs API
- Tests: unit tests for check_main, project tests for libs/mixed
Fixes #7506
Diffstat (limited to 'tests/modules/language/ascendc/test.lua')
| -rw-r--r-- | tests/modules/language/ascendc/test.lua | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/tests/modules/language/ascendc/test.lua b/tests/modules/language/ascendc/test.lua new file mode 100644 index 000000000..507bff9ec --- /dev/null +++ b/tests/modules/language/ascendc/test.lua @@ -0,0 +1,34 @@ +import("core.language.language") + +function _write_asc_file(name, source) + local sourcefile = os.tmpfile(name) .. ".asc" + io.writefile(sourcefile, source) + return sourcefile +end + +function test_check_main(t) + local instance = language.load("ascendc") + t:require(instance) + + local check_main = instance:get("check_main") + t:require(check_main) + + local mainfile = _write_asc_file("ascendc_check_main", [[ +#include "acl/acl.h" + +int32_t main(int argc, char const *argv[]) { + return 0; +} +]]) + t:are_equal(check_main(mainfile), true) + + local commentfile = _write_asc_file("ascendc_check_main_comment", [[ +// int main() { return 0; } +/* +int main() { return 1; } +*/ +__global__ __vector__ void kernel() { +} +]]) + t:are_equal(check_main(commentfile), false) +end |
