Update on "Add beginnings of torch::stable::accelerator"

mikaylagawarecki · mikaylagawarecki · commit 5e8e1588d44f · 2025-08-13T10:56:59.000-07:00
Adds - `torch::stable::accelerator::DeviceGuard`: `std::unique_ptr` to `DeviceGuardOpauqe` mostly copied from the below (but made generic) https://github.com/pytorch/pytorch/blob/50eac811a68e63e96ad56c11c983bfe298a0bb8a/torch/csrc/inductor/aoti_runtime/utils_cuda.h#L30-L46 - constructor `DeviceGuard(DeviceIndex)` (**this matches aoti but defers from the actual c10 DeviceGuard constructor that takes in device**) - `set_index(DeviceIndex)` - `torch::stable::accelerator::Stream`: `std::shared_ptr` to `StreamOpaque` - constructor `Stream(StreamHandle stream)` (similar to torch::stable::Tensor) - `id() -> StreamId` - `getCurrentStream(DeviceIndex device_index) -> stable::accelerator::Stream` [ghstack-poisoned]
diff --git a/torch/csrc/stable/tensor.h b/torch/csrc/stable/tensor.h
@@ -8,10 +8,10 @@
 
 #include <torch/csrc/stable/accelerator.h>
 
-using DeviceIndex = torch::stable::accelerator::DeviceIndex;
-
 namespace torch::stable {
 
+using DeviceIndex = torch::stable::accelerator::DeviceIndex;
+
 // The torch::stable::Tensor class is a highlevel C++ wrapper around
 // the C shim Tensor APIs. We've modeled this class after TensorBase, as custom
 // op kernels only really need to interact with Tensor metadata (think sizes,
