[SYCLomatic] Support more python script migration rules (#2750)

Signed-off-by: Jiang, Zhiwei <[email protected]>

Author: zhiweij1

Diff for: clang/test/dpct/python_migration/case_006/expected.py

@@ -56,3 +56,8 @@
 
 torch.xpu.set_stream(st)
 xpu.set_stream(st)
+
+if torch.xpu.current_stream() != torch.xpu.current_stream():
+    print("Not default stream")
+device = torch.device('xpu')
+extra_cuda_cflags=['-ffast-math', '']

Diff for: clang/test/dpct/python_migration/case_006/input.py

@@ -56,3 +56,8 @@
 
 torch.cuda.set_stream(st)
 cuda.set_stream(st)
+
+if torch.cuda.current_stream() != torch.cuda.default_stream():
+    print("Not default stream")
+device = torch.device('cuda')
+extra_cuda_cflags=['--use_fast_math', '--allow-unsupported-compiler']

Diff for: clang/tools/dpct/extensions/python_rules/python_build_script_migration_rule_pytorch_common.yaml

@@ -247,3 +247,35 @@
   PythonSyntax: CUDA_HOME
   In: CUDA_HOME
   Out: SYCL_HOME
+
+- Rule: rule_cuda_flag_use_fast_math
+  Kind: PythonRule
+  Priority: Fallback
+  MatchMode: Partial
+  PythonSyntax: cuda_flag_use_fast_math
+  In: '--use_fast_math'
+  Out: '-ffast-math'
+
+- Rule: rule_cuda_flag_allow-unsupported-compiler
+  Kind: PythonRule
+  Priority: Fallback
+  MatchMode: Partial
+  PythonSyntax: cuda_flag_allow-unsupported-compiler
+  In: '--allow-unsupported-compiler'
+  Out: ''
+
+- Rule: rule_cuda_default_stream
+  Kind: PythonRule
+  Priority: Fallback
+  MatchMode: Full
+  PythonSyntax: cuda_default_stream
+  In: torch.cuda.default_stream
+  Out: torch.xpu.current_stream
+
+- Rule: rule_torch_device_cuda
+  Kind: PythonRule
+  Priority: Fallback
+  MatchMode: Full
+  PythonSyntax: torch_device_cuda
+  In: torch.device('cuda')
+  Out: torch.device('xpu')