RuntimeError: torch._scaled_mm is only supported on devices with compute capability >= 9.0)

### 🚀 The feature, motivation and pitch

Possible to add support for the NVIDIA Ada Lovelace cards, which have a compute capability of 8.9?

NVIDIA's TransformerEngine works, so the feature is supported in the hardware.

Thanks!

@drisspg 

### Alternatives

_No response_

### Additional context

_No response_

cc @yanbing-j @vkuzo @albanD @kadeng