Fix Proton tests when PyTorch is installed with `pytorch_mode==wheels`#5381

anmyachev merged 2 commits intomainintel/intel-xpu-backend-for-triton:mainfrom

amyachev/fix-proton-testsintel/intel-xpu-backend-for-triton:amyachev/fix-proton-testsCopy head branch name to clipboard