Skip to content

[Benchmarks] Enable triton_kernels_benchmark tests to run on Nvidia device #7837

Description

@dudilester

Describe the bug

Intel triton_kernels_benchmark test scripts currently work only on XPU devices since they contain have XPU specific code.
Modifying the scripts to run on either XPU or Nvidia GPU, will enable doing performance comparison by running the exact same benchmarks scripts on Nvidia and Intel devices.

Environment details

Triton XPU
Nvidia device - RTX 3060 Ti

Metadata

Metadata

Assignees

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions