Skip to content

Commit

Permalink
Clean up stream_executor_pimpl.h include files.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 605001572
  • Loading branch information
klucke authored and copybara-github committed Feb 7, 2024
1 parent 2130bc5 commit 6ef18ff
Showing 1 changed file with 4 additions and 13 deletions.
17 changes: 4 additions & 13 deletions xla/stream_executor/stream_executor_pimpl.h
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ limitations under the License.
#define XLA_STREAM_EXECUTOR_STREAM_EXECUTOR_PIMPL_H_

#include <atomic>
#include <cstddef>
#include <cstdint>
#include <memory>
#include <optional>
Expand All @@ -43,10 +44,10 @@ limitations under the License.
#include "xla/stream_executor/kernel_spec.h"
#include "xla/stream_executor/launch_dim.h"
#include "xla/stream_executor/module_spec.h"
#include "xla/stream_executor/numeric_options.h"
#include "xla/stream_executor/platform.h"
#include "xla/stream_executor/platform/port.h"
#include "tsl/protobuf/dnn.pb.h"
#include "tsl/platform/errors.h"
#include "tsl/platform/logging.h"
#include "tsl/platform/statusor.h"

namespace stream_executor {

Expand All @@ -56,11 +57,6 @@ namespace internal {
class StreamExecutorInterface;
} // namespace internal

// Forward declaration of private friend class.
template <typename BeginCallT, typename CompleteCallT, typename ReturnT,
typename... BeginArgsT>
class ScopedTracer;

// A StreamExecutor manages a single device, in terms of executing work (kernel
// launches) and memory management (allocation/deallocation, memory copies to
// and from the device). It is conceptually the "handle" for a device -- Stream
Expand Down Expand Up @@ -373,15 +369,10 @@ class StreamExecutor {
Stream* FindAllocatedStream(void* gpu_stream);

private:
template <typename BeginCallT, typename CompleteCallT, typename ReturnT,
typename... BeginArgsT>
friend class ScopedTracer;
friend class Event;
friend class Stream;
template <typename... Params>
friend class TypedKernel;
template <typename... Args>
friend struct ThenBlasImpl;

// Synchronously allocates size bytes on the underlying platform and returns
// a DeviceMemoryBase representing that allocation. In the case of failure,
Expand Down

0 comments on commit 6ef18ff

Please sign in to comment.