We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 579bfd9 commit daa116eCopy full SHA for daa116e
1 file changed
transformer_engine/plugin/core/backends/flagos/register_ops.py
@@ -132,6 +132,14 @@ def register_builtins(registry) -> None:
132
vendor=None,
133
priority=150,
134
),
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
+ ),
143
]
144
145
registry.register_many(impls)
0 commit comments