Skip to content

Latest commit

 

History

History
67 lines (43 loc) · 2.61 KB

README.md

File metadata and controls

67 lines (43 loc) · 2.61 KB
Raw

Torch XPU Operators*

===========================

Torch XPU Operators* implements PyTorch ATen operators for Intel GPU devices, aiming to agilely support PyTroch ATen operations and buffer these operations for Intel GPU upstreaming . For more details, refer to SYCL kernels for ATen Operators RFC for more details.

1. Overview

  • SYCL Implementation for XPU Operators: The Operators in this staging branch will finally be upstreamed to PyTorch for Intel GPU.

2. Requirements

Hardware Requirements

Verified Hardware Platforms:

  • Intel® Data Center GPU Max Series, Driver Version: 803

Software Requirements

  • Ubuntu 22.04, SUSE Linux Enterprise Server(SLES) 15 SP4
    • Intel® Data Center GPU Max Series
  • Intel® oneAPI Base Toolkit 2024.0

Install Intel GPU Drivers

OS Intel GPU Install Intel GPU Driver
Ubuntu 22.04, SLES 15 SP4 Intel® Data Center GPU Max Series Refer to the Installation Guides for latest driver installation. If install the verified Intel® Data Center GPU Max Series/Intel® Data Center GPU Flex Series 803, please append the specific version after components, such as sudo apt-get install intel-opencl-icd==23.43.27642.38-803~22.04

Install oneAPI Base Toolkit Packages

Need to install components of Intel® oneAPI Base Toolkit:

  • Intel® oneAPI DPC++ Compiler
  • Intel® oneAPI Math Kernel Library (oneMKL)
  • Intel® oneAPI Threading Building Blocks (TBB), dependency of DPC++ Compiler.
wget https://registrationcenter-download.intel.com/akdlm//IRC_NAS/20f4e6a1-6b0b-4752-b8c1-e5eacba10e01/l_BaseKit_p_2024.0.0.49564.sh
# 2 components are necessary: DPC++/C++ Compiler and oneMKL
sudo sh l_BaseKit_p_2024.0.0.49564.sh

# Source OneAPI environment
source /opt/intel/oneapi/compiler/2024.0/env/vars.sh

3. Build

Need to built this project as a submodule of PyTorch, after install Intel GPU Driver and oneAPI Base Toolkit Packages.

# Setup PyTorch source project. torch-xpu-ops is included by default.
python setup.py install

4. Security

See Intel's Security Center for information on how to report a potential security issue or vulnerability.

See also: Security Policy

5. License

Apache License 2.0