pytorch/tools/setup_helpers
Yu, Guangye 8cda774a03 Add torch.xpu.get_arch_list and torch.xpu.get_gencode_flags for XPU (#137773)
# Motivation
Add `torch.xpu.get_arch_list()` and `torch.xpu.get_gencode_flags()` methods that return architecture list and AOT flags to preserve what flags PyTorch XPU was built with.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/137773
Approved by: https://github.com/EikanWang, https://github.com/albanD
2024-10-18 02:28:08 +00:00
..
__init__.py
BUILD.bazel
build.bzl
cmake.py
cmake_utils.py
env.py
gen.py
gen_unboxing.py
gen_version_header.py
generate_code.py
generate_linker_script.py