mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
a9d9077f12
This reverts commit 7c556428c7.
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
|