"""第 01 课补全练习：选择可用设备。修改 solve，保持下方检查不变。"""
import math, io, argparse
import torch
from torch import nn
from torch.nn import functional as F
torch.set_num_threads(2)
torch.manual_seed(26)

def solve(has_mps, has_cuda):
    raise NotImplementedError("请补全：选择可用设备；参考答案见 solution.py")

assert solve(True,False)=='mps'
assert solve(False,False)=='cpu'
print("本课补全练习通过。")