Skip to content

intel/torch-xpu-ops

This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Folders and files

NameName
Last commit message
Last commit date

Latest commit

3d59000 · Dec 17, 2024
Dec 17, 2024
Nov 22, 2024
Mar 26, 2024
Dec 11, 2024
Dec 4, 2024
Nov 14, 2024
Dec 11, 2024
Apr 30, 2024
Apr 30, 2024
Apr 30, 2024
Apr 30, 2024
Sep 27, 2024
Sep 26, 2024
Nov 14, 2024
Mar 8, 2024
Mar 8, 2024
Mar 7, 2024
Mar 8, 2024
Mar 7, 2024
Apr 30, 2024
Apr 30, 2024
Apr 10, 2024

Repository files navigation

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