api for new backend

[hjchen2]

No description provided.

[doombeaker]

这个分支里，comfyui 跑不起来了

  File "<frozen importlib._bootstrap_external>", line 883, in exec_module
  File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed
  File "/data/home/yaochi/onediff/ComfyUI/comfy/ops.py", line 20, in <module>
    import comfy.model_management
  File "/data/home/yaochi/onediff/ComfyUI/comfy/model_management.py", line 121, in <module>
    total_vram = get_total_memory(get_torch_device()) / (1024 * 1024)
  File "/data/home/yaochi/onediff/ComfyUI/comfy/model_management.py", line 90, in get_torch_device
    return torch.device(a)
TypeError: __init__(): incompatible constructor arguments. The following argument types are supported:
    1. oneflow._oneflow_internal.device(arg0: str)
    2. oneflow._oneflow_internal.device(type: str, index: int)
    3. oneflow._oneflow_internal.device(arg0: oneflow._oneflow_internal.device)

Invoked with: 0

表面上直接原因是 mock 机制出了点问题，貌似 fake 的那个 torch_device 没有正常 fake 成功（调用了 torch 的，所以得到了整型而不是字符串）。不过本质上还是因为 oneflow.device 没有和 torch.deice 对齐。

我觉得还是趁这个机会把 flow.device 给修了吧，不再修修补补了，会在这个 issue 里解决：https://github.com/siliconflow/sd-team/issues/350

[hjchen2]

这个分支里，comfyui 跑不起来了

  File "<frozen importlib._bootstrap_external>", line 883, in exec_module
  File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed
  File "/data/home/yaochi/onediff/ComfyUI/comfy/ops.py", line 20, in <module>
    import comfy.model_management
  File "/data/home/yaochi/onediff/ComfyUI/comfy/model_management.py", line 121, in <module>
    total_vram = get_total_memory(get_torch_device()) / (1024 * 1024)
  File "/data/home/yaochi/onediff/ComfyUI/comfy/model_management.py", line 90, in get_torch_device
    return torch.device(a)
TypeError: __init__(): incompatible constructor arguments. The following argument types are supported:
    1. oneflow._oneflow_internal.device(arg0: str)
    2. oneflow._oneflow_internal.device(type: str, index: int)
    3. oneflow._oneflow_internal.device(arg0: oneflow._oneflow_internal.device)

Invoked with: 0

表面上直接原因是 mock 机制出了点问题，貌似 fake 的那个 torch_device 没有正常 fake 成功（调用了 torch 的，所以得到了整型而不是字符串）。不过本质上还是因为 oneflow.device 没有和 torch.deice 对齐。

我觉得还是趁这个机会把 flow.device 给修了吧，不再修修补补了，会在这个 issue 里解决：siliconflow/sd-team#350

应该是有一个文件延迟导入导致的，我看看哈

[strint]

LGTM