Skip to content

Commit ae0a6eb

Browse files
committed
kernels: export ROCMProperties from top-level package
1 parent 61407ae commit ae0a6eb

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

kernels/src/kernels/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
LockedFuncRepository,
1515
LockedLayerRepository,
1616
Mode,
17+
ROCMProperties,
1718
kernelize,
1819
register_kernel_mapping,
1920
replace_kernel_forward_from_hub,
@@ -38,6 +39,7 @@
3839
"Benchmark",
3940
"CUDAProperties",
4041
"Device",
42+
"ROCMProperties",
4143
"FuncRepository",
4244
"LayerRepository",
4345
"LocalFuncRepository",

0 commit comments

Comments
 (0)