File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -124,7 +124,7 @@ else:
124
124
125
125
# Run the dpct sanity test.
126
126
if 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" ]
128
128
complete_process = run_sanity ("cudnn_sanity.cu" )
129
129
err_message = ""
130
130
if complete_process .returncode != 0 :
@@ -147,6 +147,13 @@ if not config.unsupported:
147
147
"Please make sure install the header file of nvshmem and " + \
148
148
"export nvshmem.h in CPATH.\n "
149
149
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>" ))
150
157
151
158
if (err_message ):
152
159
sys .stderr .write (err_message )
You can’t perform that action at this time.
0 commit comments