Merge pull request #371 from kmaehashi/preload-log

Use new library load debugging feature
cupy · Nov 17, 2023 · d941446 · d941446
2 parents 91adbe6 + 8d4261b
commit d941446
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/verifier/agent.py b/verifier/agent.py
@@ -17,7 +17,9 @@ def _log(self, msg):
 
     def _run(self, *cmd):
         self._log('Running command: {0}'.format(str(cmd)))
-        subprocess.check_call(cmd)
+        env = dict(os.environ)
+        env['CUPY_DEBUG_LIBRARY_LOAD'] = '1'
+        subprocess.check_call(cmd, env=env)
 
     def parse_args(self):
         parser = argparse.ArgumentParser()
@@ -79,7 +81,6 @@ def main(self):
                 'import cupy',
                 'import cupy.cuda.cudnn',
                 'cupy.show_config()',
-                'print(cupy._environment._get_preload_logs())'
             ])
         ]
         self._run(*cmdline)