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" ]
127+ skipped_cases = ["cudnn_sanity.cu" , "nccl_sanity.cu" , "nvshmem.cu" ]
128128 complete_process = run_sanity ("cudnn_sanity.cu" )
129129 err_message = ""
130130 if complete_process .returncode != 0 :
@@ -140,6 +140,14 @@ if not config.unsupported:
140140 "Please make sure install the header file of NCCL and " + \
141141 "export nccl.h in CPATH.\n "
142142 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+
143151 if (err_message ):
144152 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