mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
a9d9077f12
This reverts commit 7c556428c74a79c6d9c272826344a0828d3f66f5. Reverted https://github.com/pytorch/pytorch/pull/119639 on behalf of https://github.com/kit1980 due to breaking internal builds, see D54286923 ([comment](https://github.com/pytorch/pytorch/pull/119639#issuecomment-1969634480))
12 lines
185 B
C++
12 lines
185 B
C++
#pragma once
|
|
|
|
#include <c10/cuda/CUDAMacros.h>
|
|
#include <bitset>
|
|
#include <cstddef>
|
|
|
|
namespace torch {
|
|
|
|
using device_set = std::bitset<C10_COMPILE_TIME_MAX_GPUS>;
|
|
|
|
} // namespace torch
|