File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -124,7 +124,7 @@ else:
124124
125125# Run the dpct sanity test.
126126if not config .unsupported :
127- skipped_cases = ["cudnn_sanity.cu" , "nccl_sanity.cu" , "nvshmem.cu" ]
127+ skipped_cases = ["cudnn_sanity.cu" , "nccl_sanity.cu" , "nvshmem.cu" , "cutensor.cu" ]
128128 complete_process = run_sanity ("cudnn_sanity.cu" )
129129 err_message = ""
130130 if complete_process .returncode != 0 :
@@ -147,6 +147,13 @@ if not config.unsupported:
147147 "Please make sure install the header file of nvshmem and " + \
148148 "export nvshmem.h in CPATH.\n "
149149 skipped_cases .extend (get_skipped_cases_with_string ("<nccl.h>" ))
150+ complete_process = run_sanity ("cutensor.cu" )
151+ if complete_process .returncode != 0 :
152+ if "'cutensor.h' file not found" in complete_process .stdout :
153+ err_message += "'cutensor.h' header file not found in platform. " + \
154+ "Please make sure install the header file of cuTENSOR and " + \
155+ "export cutensor.h in CPATH.\n "
156+ skipped_cases .extend (get_skipped_cases_with_string ("<cutensor.h>" ))
150157
151158 if (err_message ):
152159 sys .stderr .write (err_message )
You can’t perform that action at this time.
0 commit comments