oneDNN error "Not support device type cuda"

jit
1

I’m following the guide, after adding the line torch.jit.enable_onednn_fusion(True)
it creates the following error:
RuntimeError: The following operation failed in the TorchScript interpreter. Traceback of TorchScript (most recent call last): RuntimeError: Not support device type cuda

My model is compiled with torch.jit.script, stack trace shows basically nothing
Is oneDNN cpu only or did I not install something required?

2

Oh it was listed under CPU specific optimizations didn’t see the category