Does DeepSpeed 0.9.x support PyTorch2.x+cu121

[wengyao04]

Hi Does DeepSpeed 0.9.1 support PyTorch2.x+cud12 ? We want to run DeepSpeed with torch on H100GPU with 12.1, we try to use nightly build torch2.1.0+cu121 and build DeepSpeed wheel, however we see the following error

/opt/bb/lib/python3.8/site-packages/torch/include/torch/csrc/api/include/torch/all.h:4:2: error: #error C++17 or later compatible compiler is required to use PyTorch.
    4 | #error C++17 or later compatible compiler is required to use PyTorch.
      |  ^~~~~

we use gcc from devtools-12 which can compile C++17 code. I think the issue comes from -std=c++14, when I run

$PYTHON setup.py bdist_wheel

/opt/rh/devtoolset-12/root/usr/bin/gcc -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -m64 -march=westmere -fno-semantic-interposition -fdebug-prefix-map=/tmp/python3.8-3.8.17-0=/opt/bb/src/python/python3.8 -g3 -frecord-gcc-switches -D_FORTIFY_SOURCE=2 -fstack-clash-protection -fstack-protector-strong -Wformat -Werror=format-security -I/opt/bb/include/ -fPIC -Icsrc/aio/py_lib -Icsrc/aio/common -I/opt/bb/lib/python3.8/site-packages/torch/include -I/opt/bb/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/opt/bb/lib/python3.8/site-packages/torch/include/TH -I/opt/bb/lib/python3.8/site-packages/torch/include/THC -I/opt/bb/include/python3.8 -c csrc/aio/common/deepspeed_aio_utils.cpp -o build/temp.linux-x86_64-cpython-38/csrc/aio/common/deepspeed_aio_utils.o -g -Wall -O0 -std=c++14 -shared -fPIC -Wno-reorder -march=native -fopenmp -D__SCALAR__ -laio -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE=\"_gcc\" -DPYBIND11_STDLIB=\"_libstdcpp\" -DPYBIND11_BUILD_ABI=\"_cxxabi1011\" -DTORCH_EXTENSION_NAME=async_io_op -D_GLIBCXX_USE_CXX11_ABI=0

How could I override -std=c++14 with std=c++17 ?