File tree Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Expand file tree Collapse file tree 1 file changed +10
-2
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" ]
127
+ skipped_cases = ["cudnn_sanity.cu" , "nccl_sanity.cu" , "nvshmem.cu" ]
128
128
complete_process = run_sanity ("cudnn_sanity.cu" )
129
129
err_message = ""
130
130
if complete_process .returncode != 0 :
@@ -140,6 +140,14 @@ if not config.unsupported:
140
140
"Please make sure install the header file of NCCL and " + \
141
141
"export nccl.h in CPATH.\n "
142
142
skipped_cases .extend (get_skipped_cases_with_string ("<nccl.h>" ))
143
+ complete_process = run_sanity ("nvshmem.cu" )
144
+ if complete_process .returncode != 0 :
145
+ if "'nvshmem.h' file not found" in complete_process .stdout :
146
+ err_message += "'nvshmem.h' header file not found in platform. " + \
147
+ "Please make sure install the header file of nvshmem and " + \
148
+ "export nvshmem.h in CPATH.\n "
149
+ skipped_cases .extend (get_skipped_cases_with_string ("<nccl.h>" ))
150
+
143
151
if (err_message ):
144
152
sys .stderr .write (err_message )
145
- config .excludes = skipped_cases
153
+ config .excludes = skipped_cases
You can’t perform that action at this time.
0 commit comments