diff --git a/clang/test/dpct/lit.local.cfg b/clang/test/dpct/lit.local.cfg index aad06614afc6..45152ee265eb 100644 --- a/clang/test/dpct/lit.local.cfg +++ b/clang/test/dpct/lit.local.cfg @@ -124,7 +124,7 @@ else: # Run the dpct sanity test. if not config.unsupported: - skipped_cases = ["cudnn_sanity.cu", "nccl_sanity.cu", "nvshmem.cu"] + skipped_cases = ["cudnn_sanity.cu", "nccl_sanity.cu", "nvshmem.cu", "cutensor.cu"] complete_process = run_sanity("cudnn_sanity.cu") err_message = "" if complete_process.returncode != 0: @@ -147,6 +147,13 @@ if not config.unsupported: "Please make sure install the header file of nvshmem and " + \ "export nvshmem.h in CPATH.\n" skipped_cases.extend(get_skipped_cases_with_string("")) + complete_process = run_sanity("cutensor.cu") + if complete_process.returncode != 0: + if "'cutensor.h' file not found" in complete_process.stdout: + err_message += "'cutensor.h' header file not found in platform. " + \ + "Please make sure install the header file of cuTENSOR and " + \ + "export cutensor.h in CPATH.\n" + skipped_cases.extend(get_skipped_cases_with_string("")) if (err_message): sys.stderr.write(err_message)