ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

add parsing grouped conv fwd instances

Browse Source
This commit is contained in:
Max Podkorytov
2024-10-18 17:09:12 +00:00
parent 7d576f1748
commit eda5938386
5 changed files with 271 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/ck4inductor/universal_gemm/gen_instances.py
View File

@@ -1,7 +1,10 @@
# SPDX-License-Identifier: MIT
# Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.
import logging
import os
import subprocess
from dataclasses import fields, replace
from dataclasses import replace
from functools import lru_cache, partial
from typing import List

3
python/ck4inductor/universal_gemm/op.py
View File

@@ -1,3 +1,6 @@
# SPDX-License-Identifier: MIT
# Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.
from dataclasses import asdict, dataclass
from typing import Optional, Tuple