[Diffusion] [NPU] Fix CI run

[Makcum888e]

Motivation

sometimes CI test still not runned
and NPU CI now broken

Example
https://github.com/sgl-project/sglang/pull/17807/changes

Modifications

==true to make compare by boolean
replace torch.cuda with torch.get_device_module()

Accuracy Tests

Benchmarking and Profiling

Checklist

• Format your code according to the Format code with pre-commit.
• Add unit tests according to the Run and add unit tests.
• Update documentation according to Write documentations.
• Provide accuracy and speed benchmark results according to Test the accuracy and Benchmark the speed.
• Follow the SGLang code style guidance.

Review Process

1. Ping Merge Oncalls to start the PR flow. See the PR Merge Process.
2. Get approvals from CODEOWNERS and other reviewers.
3. Trigger CI tests with comments or contact authorized users to do so.
   • /tag-run-ci-label, /rerun-failed-ci, /tag-and-rerun-ci
4. After green CI and required approvals, ask Merge Oncalls to merge.

[gemini-code-assist]

Note

Gemini is unable to generate a summary for this pull request due to the file types involved not being currently supported.

[ping1jing2]

please add more details in the description

[Makcum888e]

please add more details in the description

added

[ping1jing2]

/tag-and-rerun-ci