Skip to content

Commit

Permalink
Remove unused MemoryTypeString function.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 713724579
  • Loading branch information
klucke authored and Google-ML-Automation committed Jan 9, 2025
1 parent 8cb97ad commit defdfa7
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 13 deletions.
2 changes: 2 additions & 0 deletions xla/stream_executor/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -399,9 +399,11 @@ cc_library(
":allocator_stats",
":blas",
":command_buffer",
":device_description",
":device_memory",
":dnn",
":event",
":event_based_timer",
":fft",
":kernel",
":kernel_spec",
Expand Down
15 changes: 2 additions & 13 deletions xla/stream_executor/stream_executor.h
Original file line number Diff line number Diff line change
Expand Up @@ -31,9 +31,11 @@ limitations under the License.
#include "xla/stream_executor/allocator_stats.h"
#include "xla/stream_executor/blas.h"
#include "xla/stream_executor/command_buffer.h"
#include "xla/stream_executor/device_description.h"
#include "xla/stream_executor/device_memory.h"
#include "xla/stream_executor/dnn.h"
#include "xla/stream_executor/event.h"
#include "xla/stream_executor/event_based_timer.h"
#include "xla/stream_executor/fft.h"
#include "xla/stream_executor/kernel.h"
#include "xla/stream_executor/kernel_spec.h"
Expand All @@ -47,19 +49,6 @@ namespace stream_executor {
// Identifies the memory space where an allocation resides.
enum class MemoryType { kDevice = 0, kUnified, kCollective, kHost = 5 };

inline std::string MemoryTypeString(MemoryType memory_type) {
switch (memory_type) {
case MemoryType::kDevice:
return "device";
case MemoryType::kUnified:
return "unified";
case MemoryType::kCollective:
return "collective";
case MemoryType::kHost:
return "host";
}
}

/// The StreamExecutor is a single-device abstraction for:
//
// * Loading/launching data-parallel-kernels
Expand Down

0 comments on commit defdfa7

Please sign in to comment.