Minor cleanup: remove unused constructions
PiperOrigin-RevId: 160736783
This commit is contained in:
parent
9999dd321a
commit
818b768cda
@ -25,7 +25,6 @@ limitations under the License.
|
|||||||
#include "tensorflow/compiler/tf2xla/dump_graph.h"
|
#include "tensorflow/compiler/tf2xla/dump_graph.h"
|
||||||
#include "tensorflow/core/common_runtime/function.h"
|
#include "tensorflow/core/common_runtime/function.h"
|
||||||
#include "tensorflow/core/common_runtime/optimization_registry.h"
|
#include "tensorflow/core/common_runtime/optimization_registry.h"
|
||||||
#include "tensorflow/core/framework/attr_value.pb.h"
|
|
||||||
#include "tensorflow/core/framework/function.h"
|
#include "tensorflow/core/framework/function.h"
|
||||||
#include "tensorflow/core/framework/graph_def_util.h"
|
#include "tensorflow/core/framework/graph_def_util.h"
|
||||||
#include "tensorflow/core/framework/node_def_builder.h"
|
#include "tensorflow/core/framework/node_def_builder.h"
|
||||||
|
@ -18,7 +18,6 @@ limitations under the License.
|
|||||||
|
|
||||||
#include "tensorflow/compiler/plugin/executor/compiler.h"
|
#include "tensorflow/compiler/plugin/executor/compiler.h"
|
||||||
#include "tensorflow/compiler/plugin/executor/executable.h"
|
#include "tensorflow/compiler/plugin/executor/executable.h"
|
||||||
|
|
||||||
#include "tensorflow/compiler/xla/service/algebraic_simplifier.h"
|
#include "tensorflow/compiler/xla/service/algebraic_simplifier.h"
|
||||||
#include "tensorflow/compiler/xla/service/flatten_call_graph.h"
|
#include "tensorflow/compiler/xla/service/flatten_call_graph.h"
|
||||||
#include "tensorflow/compiler/xla/service/hlo_constant_folding.h"
|
#include "tensorflow/compiler/xla/service/hlo_constant_folding.h"
|
||||||
@ -30,18 +29,15 @@ limitations under the License.
|
|||||||
#include "tensorflow/compiler/xla/service/inliner.h"
|
#include "tensorflow/compiler/xla/service/inliner.h"
|
||||||
#include "tensorflow/compiler/xla/service/reshape_mover.h"
|
#include "tensorflow/compiler/xla/service/reshape_mover.h"
|
||||||
#include "tensorflow/compiler/xla/status_macros.h"
|
#include "tensorflow/compiler/xla/status_macros.h"
|
||||||
|
#include "tensorflow/core/lib/core/errors.h"
|
||||||
#include "tensorflow/stream_executor/lib/initialize.h"
|
#include "tensorflow/stream_executor/lib/initialize.h"
|
||||||
#include "tensorflow/stream_executor/lib/strcat.h"
|
#include "tensorflow/stream_executor/lib/strcat.h"
|
||||||
|
|
||||||
#include "tensorflow/core/lib/core/errors.h"
|
namespace xla {
|
||||||
|
namespace executorplugin {
|
||||||
|
|
||||||
namespace se = ::perftools::gputools;
|
namespace se = ::perftools::gputools;
|
||||||
namespace sep = ::perftools::gputools::executorplugin;
|
namespace sep = ::perftools::gputools::executorplugin;
|
||||||
namespace port = ::perftools::gputools::port;
|
|
||||||
|
|
||||||
namespace xla {
|
|
||||||
namespace executorplugin {
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Run optimization passes on the module. The graph is transformed by
|
* Run optimization passes on the module. The graph is transformed by
|
||||||
@ -111,12 +107,11 @@ ExecutorCompiler::ShapeSizeBytesFunction() const {
|
|||||||
return ExecutorExecutable::ShapeSizeBytes;
|
return ExecutorExecutable::ShapeSizeBytes;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
} // namespace executorplugin
|
|
||||||
} // namespace xla
|
|
||||||
|
|
||||||
REGISTER_MODULE_INITIALIZER(executor_compiler, {
|
REGISTER_MODULE_INITIALIZER(executor_compiler, {
|
||||||
xla::Compiler::RegisterCompilerFactory(sep::kExecutorPlatformId, []() {
|
xla::Compiler::RegisterCompilerFactory(sep::kExecutorPlatformId, []() {
|
||||||
return xla::MakeUnique<xla::executorplugin::ExecutorCompiler>();
|
return xla::MakeUnique<xla::executorplugin::ExecutorCompiler>();
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
|
} // namespace executorplugin
|
||||||
|
} // namespace xla
|
||||||
|
@ -15,18 +15,16 @@ limitations under the License.
|
|||||||
|
|
||||||
#include "tensorflow/compiler/plugin/executor/executable.h"
|
#include "tensorflow/compiler/plugin/executor/executable.h"
|
||||||
#include "tensorflow/compiler/plugin/executor/executor.h"
|
#include "tensorflow/compiler/plugin/executor/executor.h"
|
||||||
|
|
||||||
#include "tensorflow/compiler/xla/service/hlo_evaluator.h"
|
|
||||||
|
|
||||||
#include "tensorflow/compiler/xla/literal_util.h"
|
#include "tensorflow/compiler/xla/literal_util.h"
|
||||||
|
#include "tensorflow/compiler/xla/service/hlo_evaluator.h"
|
||||||
#include "tensorflow/compiler/xla/shape_util.h"
|
#include "tensorflow/compiler/xla/shape_util.h"
|
||||||
|
|
||||||
namespace se = ::perftools::gputools;
|
|
||||||
namespace sep = ::perftools::gputools::executorplugin;
|
|
||||||
|
|
||||||
namespace xla {
|
namespace xla {
|
||||||
namespace executorplugin {
|
namespace executorplugin {
|
||||||
|
|
||||||
|
namespace se = ::perftools::gputools;
|
||||||
|
namespace sep = ::perftools::gputools::executorplugin;
|
||||||
|
|
||||||
ExecutorExecutable::ExecutorExecutable(std::unique_ptr<HloModule> hlo_module)
|
ExecutorExecutable::ExecutorExecutable(std::unique_ptr<HloModule> hlo_module)
|
||||||
: Executable(std::move(hlo_module), ShapeSizeBytes) {}
|
: Executable(std::move(hlo_module), ShapeSizeBytes) {}
|
||||||
|
|
||||||
@ -90,15 +88,14 @@ StatusOr<se::DeviceMemoryBase> ExecutorExecutable::ExecuteOnStream(
|
|||||||
arg_literals_ptrs.push_back(arg_literals.back().get());
|
arg_literals_ptrs.push_back(arg_literals.back().get());
|
||||||
|
|
||||||
// Copy in the data from the stream_executor buffers
|
// Copy in the data from the stream_executor buffers
|
||||||
void* buffer = arg_literals.back().get()->MutableInternalData();
|
void* buffer = arg_literals.back()->MutableInternalData();
|
||||||
memcpy(buffer, arguments[p].opaque(),
|
memcpy(buffer, arguments[p].opaque(),
|
||||||
ShapeUtil::ByteSizeOf(param->shape()));
|
ShapeUtil::ByteSizeOf(param->shape()));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute the graph using the evaluator
|
// Execute the graph using the evaluator
|
||||||
HloEvaluator evaluator;
|
HloEvaluator evaluator;
|
||||||
std::unique_ptr<Literal> output;
|
TF_ASSIGN_OR_RETURN(std::unique_ptr<Literal> output,
|
||||||
TF_ASSIGN_OR_RETURN(output,
|
|
||||||
evaluator.Evaluate(computation, arg_literals_ptrs));
|
evaluator.Evaluate(computation, arg_literals_ptrs));
|
||||||
|
|
||||||
// Copy the result into the return buffer
|
// Copy the result into the return buffer
|
||||||
|
@ -14,14 +14,12 @@ limitations under the License.
|
|||||||
==============================================================================*/
|
==============================================================================*/
|
||||||
|
|
||||||
#include "tensorflow/compiler/plugin/executor/executor.h"
|
#include "tensorflow/compiler/plugin/executor/executor.h"
|
||||||
#include "tensorflow/compiler/plugin/executor/platform_id.h"
|
|
||||||
|
|
||||||
#include "tensorflow/compiler/xla/status_macros.h"
|
|
||||||
|
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
namespace se = ::perftools::gputools;
|
#include "tensorflow/compiler/plugin/executor/platform_id.h"
|
||||||
|
#include "tensorflow/compiler/xla/status_macros.h"
|
||||||
|
|
||||||
namespace perftools {
|
namespace perftools {
|
||||||
namespace gputools {
|
namespace gputools {
|
||||||
@ -37,10 +35,7 @@ ExecutorExecutor::ExecutorExecutor(const PluginConfig &plugin_config)
|
|||||||
|
|
||||||
ExecutorExecutor::~ExecutorExecutor() {}
|
ExecutorExecutor::~ExecutorExecutor() {}
|
||||||
|
|
||||||
void *ExecutorExecutor::Allocate(uint64 size) {
|
void *ExecutorExecutor::Allocate(uint64 size) { return new char[size]; }
|
||||||
void *buf = new char[size];
|
|
||||||
return buf;
|
|
||||||
}
|
|
||||||
|
|
||||||
void *ExecutorExecutor::AllocateSubBuffer(DeviceMemoryBase *parent,
|
void *ExecutorExecutor::AllocateSubBuffer(DeviceMemoryBase *parent,
|
||||||
uint64 offset_bytes,
|
uint64 offset_bytes,
|
||||||
@ -126,8 +121,7 @@ DeviceDescription *ExecutorExecutor::PopulateDeviceDescription() const {
|
|||||||
builder.set_device_memory_size(static_cast<uint64>(4) * 1024 * 1024 * 1024);
|
builder.set_device_memory_size(static_cast<uint64>(4) * 1024 * 1024 * 1024);
|
||||||
builder.set_clock_rate_ghz(static_cast<float>(CLOCKS_PER_SEC) / 1e9);
|
builder.set_clock_rate_ghz(static_cast<float>(CLOCKS_PER_SEC) / 1e9);
|
||||||
|
|
||||||
auto built = builder.Build();
|
return builder.Build().release();
|
||||||
return built.release();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace executorplugin
|
} // namespace executorplugin
|
||||||
|
@ -357,7 +357,7 @@ TEST_F(HloRematerializationTest, InstructionRematerializedMultipleTimes) {
|
|||||||
/*dimension=*/0));
|
/*dimension=*/0));
|
||||||
builder.AddInstruction(HloInstruction::CreateSlice(
|
builder.AddInstruction(HloInstruction::CreateSlice(
|
||||||
vec1024_shape_, concat, /*start_indices=*/{0},
|
vec1024_shape_, concat, /*start_indices=*/{0},
|
||||||
/*limit_indices=*/{1024}, /*slices=*/{1}));
|
/*limit_indices=*/{1024}, /*strides=*/{1}));
|
||||||
subcomputation = module->AddEmbeddedComputation(builder.Build());
|
subcomputation = module->AddEmbeddedComputation(builder.Build());
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -473,7 +473,7 @@ TEST_P(IndirectUseTest, IndirectUseNotRematerialized) {
|
|||||||
/*dimension=*/0));
|
/*dimension=*/0));
|
||||||
builder.AddInstruction(HloInstruction::CreateSlice(
|
builder.AddInstruction(HloInstruction::CreateSlice(
|
||||||
vec1024_shape_, concat, /*start_indices=*/{0},
|
vec1024_shape_, concat, /*start_indices=*/{0},
|
||||||
/*limit_indices=*/{1024}, /*slices=*/{1}));
|
/*limit_indices=*/{1024}, /*strides=*/{1}));
|
||||||
subcomputation = module->AddEmbeddedComputation(builder.Build());
|
subcomputation = module->AddEmbeddedComputation(builder.Build());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -19,7 +19,6 @@
|
|||||||
#include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h"
|
#include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h"
|
||||||
#include "tensorflow/contrib/boosted_trees/resources/decision_tree_ensemble_resource.h"
|
#include "tensorflow/contrib/boosted_trees/resources/decision_tree_ensemble_resource.h"
|
||||||
#include "tensorflow/core/framework/op_kernel.h"
|
#include "tensorflow/core/framework/op_kernel.h"
|
||||||
#include "tensorflow/core/framework/resource_handle.pb.h"
|
|
||||||
#include "tensorflow/core/framework/resource_mgr.h"
|
#include "tensorflow/core/framework/resource_mgr.h"
|
||||||
#include "tensorflow/core/framework/tensor.h"
|
#include "tensorflow/core/framework/tensor.h"
|
||||||
#include "tensorflow/core/framework/tensor_shape.h"
|
#include "tensorflow/core/framework/tensor_shape.h"
|
||||||
|
@ -18,7 +18,6 @@
|
|||||||
#include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h"
|
#include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h"
|
||||||
#include "tensorflow/contrib/boosted_trees/resources/decision_tree_ensemble_resource.h"
|
#include "tensorflow/contrib/boosted_trees/resources/decision_tree_ensemble_resource.h"
|
||||||
#include "tensorflow/core/framework/op_kernel.h"
|
#include "tensorflow/core/framework/op_kernel.h"
|
||||||
#include "tensorflow/core/framework/resource_handle.pb.h"
|
|
||||||
#include "tensorflow/core/framework/resource_mgr.h"
|
#include "tensorflow/core/framework/resource_mgr.h"
|
||||||
#include "tensorflow/core/framework/tensor.h"
|
#include "tensorflow/core/framework/tensor.h"
|
||||||
#include "tensorflow/core/framework/tensor_shape.h"
|
#include "tensorflow/core/framework/tensor_shape.h"
|
||||||
|
@ -19,7 +19,6 @@
|
|||||||
#include "tensorflow/contrib/tensor_forest/kernels/v4/input_data.h"
|
#include "tensorflow/contrib/tensor_forest/kernels/v4/input_data.h"
|
||||||
#include "tensorflow/contrib/tensor_forest/proto/tensor_forest_params.pb.h"
|
#include "tensorflow/contrib/tensor_forest/proto/tensor_forest_params.pb.h"
|
||||||
#include "tensorflow/core/framework/op_kernel.h"
|
#include "tensorflow/core/framework/op_kernel.h"
|
||||||
#include "tensorflow/core/framework/resource_handle.pb.h"
|
|
||||||
#include "tensorflow/core/framework/resource_mgr.h"
|
#include "tensorflow/core/framework/resource_mgr.h"
|
||||||
#include "tensorflow/core/framework/tensor.h"
|
#include "tensorflow/core/framework/tensor.h"
|
||||||
#include "tensorflow/core/framework/tensor_shape.h"
|
#include "tensorflow/core/framework/tensor_shape.h"
|
||||||
|
@ -22,7 +22,6 @@
|
|||||||
#include "tensorflow/contrib/tensor_forest/kernels/v4/params.h"
|
#include "tensorflow/contrib/tensor_forest/kernels/v4/params.h"
|
||||||
#include "tensorflow/contrib/tensor_forest/proto/fertile_stats.pb.h"
|
#include "tensorflow/contrib/tensor_forest/proto/fertile_stats.pb.h"
|
||||||
#include "tensorflow/core/framework/op_kernel.h"
|
#include "tensorflow/core/framework/op_kernel.h"
|
||||||
#include "tensorflow/core/framework/resource_handle.pb.h"
|
|
||||||
#include "tensorflow/core/framework/resource_mgr.h"
|
#include "tensorflow/core/framework/resource_mgr.h"
|
||||||
#include "tensorflow/core/framework/tensor.h"
|
#include "tensorflow/core/framework/tensor.h"
|
||||||
#include "tensorflow/core/framework/tensor_shape.h"
|
#include "tensorflow/core/framework/tensor_shape.h"
|
||||||
|
@ -18,7 +18,6 @@ limitations under the License.
|
|||||||
|
|
||||||
namespace tensorflow {
|
namespace tensorflow {
|
||||||
|
|
||||||
using shape_inference::DimensionHandle;
|
|
||||||
using shape_inference::InferenceContext;
|
using shape_inference::InferenceContext;
|
||||||
using shape_inference::ShapeHandle;
|
using shape_inference::ShapeHandle;
|
||||||
|
|
||||||
|
@ -16,7 +16,6 @@ limitations under the License.
|
|||||||
#include "tensorflow/core/framework/common_shape_fns.h"
|
#include "tensorflow/core/framework/common_shape_fns.h"
|
||||||
#include "tensorflow/core/framework/op.h"
|
#include "tensorflow/core/framework/op.h"
|
||||||
#include "tensorflow/core/framework/op_kernel.h"
|
#include "tensorflow/core/framework/op_kernel.h"
|
||||||
#include "tensorflow/core/framework/resource_handle.pb.h"
|
|
||||||
#include "tensorflow/core/framework/resource_mgr.h"
|
#include "tensorflow/core/framework/resource_mgr.h"
|
||||||
#include "tensorflow/core/lib/core/status.h"
|
#include "tensorflow/core/lib/core/status.h"
|
||||||
#include "tensorflow/core/public/version.h"
|
#include "tensorflow/core/public/version.h"
|
||||||
|
Loading…
Reference in New Issue
Block a user