Don't use cpu for tensor comparison in `python/tutorials/01-vector-add.py` #3241

anmyachev · 2025-01-23T12:27:03Z

intel-xpu-backend-for-triton/python/tutorials/01-vector-add.py

Lines 90 to 93 in d7d55b8

    
           print(output_torch.cpu()) 
        
           print(output_triton.cpu()) 
        
           print(f'The maximum difference between torch and triton is ' 
        
                 f'{torch.max(torch.abs(output_torch.cpu() - output_triton.cpu()))}')

Part of #2030

anmyachev added the enhancement New feature or request label Jan 23, 2025

anmyachev linked a pull request Jan 23, 2025 that will close this issue

Don't use cpu for tensor comparison in python/tutorials/01-vector-add.py #3242

Merged

anmyachev closed this as completed in #3242 Jan 24, 2025

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Don't use cpu for tensor comparison in `python/tutorials/01-vector-add.py` #3241

Don't use cpu for tensor comparison in `python/tutorials/01-vector-add.py` #3241

anmyachev commented Jan 23, 2025

Don't use cpu for tensor comparison in python/tutorials/01-vector-add.py #3241

Don't use cpu for tensor comparison in python/tutorials/01-vector-add.py #3241

Comments

anmyachev commented Jan 23, 2025

Don't use cpu for tensor comparison in `python/tutorials/01-vector-add.py` #3241

Don't use cpu for tensor comparison in `python/tutorials/01-vector-add.py` #3241