update and simplify CustomOp

[zhyncs]

Motivation

Modifications

Checklist

• Format your code according to the Code Formatting with Pre-Commit.
• Add unit tests as outlined in the Running Unit Tests.
• Update documentation / docstrings / example tutorials as needed, according to Writing Documentation.
• Provide throughput / latency benchmark results and accuracy evaluation results as needed, according to Benchmark and Profiling.

[zhyncs]

ref
https://pytorch.org/docs/stable/notes/hip.html#checking-for-hip

https://github.com/pytorch/pytorch/blob/30f091da44482d55f19eccff53eb6e26aea780d4/tools/dynamo/verify_dynamo.py#L82-L142

[zhyncs]

All CIs have passed