Skip to content

Commit daa116e

Browse files
committed
add register for get_cudnn_version for flagos backend
1 parent 579bfd9 commit daa116e

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

transformer_engine/plugin/core/backends/flagos/register_ops.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,14 @@ def register_builtins(registry) -> None:
132132
vendor=None,
133133
priority=150,
134134
),
135+
OpImpl(
136+
op_name="get_cudnn_version",
137+
impl_id="default.flagos",
138+
kind=BackendImplKind.DEFAULT,
139+
fn=_bind_is_available(backend.get_cudnn_version, is_avail),
140+
vendor=None,
141+
priority=150,
142+
),
135143
]
136144

137145
registry.register_many(impls)

0 commit comments

Comments
 (0)