()
| 1391 | |
| 1392 | |
| 1393 | def build_cutlass3_src_code(): |
| 1394 | target_path = f"{paddle_binary_dir}/python/paddle/apy/matmul_pass/matmul/cutlass-3.7.0" |
| 1395 | if not os.path.exists(target_path): |
| 1396 | os.mkdir(target_path) |
| 1397 | try: |
| 1398 | cmd = ['git', 'rev-parse', 'HEAD'] |
| 1399 | git_commit = ( |
| 1400 | subprocess.Popen( |
| 1401 | cmd, |
| 1402 | stdout=subprocess.PIPE, |
| 1403 | cwd=f"{paddle_source_dir}/third_party/cutlass", |
| 1404 | ) |
| 1405 | .communicate()[0] |
| 1406 | .strip() |
| 1407 | ) |
| 1408 | except: |
| 1409 | git_commit = 'Unknown' |
| 1410 | raise Exception("obtain commit id of third_party cutlass failed") |
| 1411 | commit_id = str(git_commit.decode()) |
| 1412 | command = ( |
| 1413 | 'cd ' |
| 1414 | + f'{paddle_source_dir}/third_party/cutlass && ' |
| 1415 | + 'git checkout v3.7.0 && ' |
| 1416 | + 'cp ' |
| 1417 | + f'{paddle_source_dir}/third_party/cutlass/tools -r ' |
| 1418 | + f'{target_path} && ' |
| 1419 | + 'cp ' |
| 1420 | + f'{paddle_source_dir}/third_party/cutlass/include -r ' |
| 1421 | + f'{target_path} && ' |
| 1422 | + f'git checkout {commit_id}' |
| 1423 | ) |
| 1424 | if os.system(command) != 0: |
| 1425 | raise Exception(f"copy cutlass-3.7.0 failed, command: {command}") |
| 1426 | |
| 1427 | |
| 1428 | def get_cinn_config_jsons(): |
nothing calls this directly
no test coverage detected