Skip to content

Commit

Permalink
[XLA:GPU][NFC] Delete dead methods in `triton_fusion_emitter_parametr…
Browse files Browse the repository at this point in the history
…ized_test.cc`.

PiperOrigin-RevId: 660731941
  • Loading branch information
bchetioui authored and tensorflower-gardener committed Aug 8, 2024
1 parent a27ccbc commit d70e4b5
Showing 1 changed file with 0 additions and 14 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -52,13 +52,6 @@ struct MixTypeParams {
class MixedTypeTest : public GpuCodegenTest,
public ::testing::WithParamInterface<MixTypeParams> {
public:
se::CudaComputeCapability GetCudaComputeCapability() {
return backend()
.default_stream_executor()
->GetDeviceDescription()
.cuda_compute_capability();
}

DebugOptions GetDebugOptionsForTest() override {
DebugOptions debug_options = GpuCodegenTest::GetDebugOptionsForTest();
// We are testing Triton, remove cuBLAS fallback for these tests.
Expand Down Expand Up @@ -803,13 +796,6 @@ class TritonSoftmaxTest : public GpuCodegenTest,
debug_options.clear_xla_disable_hlo_passes();
return debug_options;
}

se::CudaComputeCapability GetCudaComputeCapability() {
return backend()
.default_stream_executor()
->GetDeviceDescription()
.cuda_compute_capability();
}
};

TEST_P(TritonSoftmaxTest, CanFuseAndEmitExactSoftmax) {
Expand Down

0 comments on commit d70e4b5

Please sign in to comment.