[CI] Support smoke test in the workflow

[can-gaa-hou]

Pass the essential PyTorch smoke testing as defined by https://github.com/pytorch/pytorch/blob/main/.ci/pytorch/smoke_test/smoke_test.py

cc @fffrog @zeshengzong