convolution_cudnn_test.cc
Google Test frameworkimports tsl/platform/path.h
SHA-2565a364b5b5f36f6aeeb2a81fd631d933b3d3e3bef186b3348e494da7778ffcc2a
MaleculeMd₂(In₂Pa)
Evidence
1/* Copyright 2017 The OpenXLA Authors.
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is …
11:42… IND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14==============================================================================*/
15
16#include <cstdint>
17#include <memory>
18#include <string>
19#include <tuple>
20#include <utility>
21
22#include <gtest/gtest.h>
23#include "absl/strings/string_view.h"
24#include "absl/strings/substitute.h"
25#include "xla/error_spec.h"
26#include "xla/hlo/ir/hlo_module.h"
27#include "xla/service/gpu/tests/hlo_pjrt_gpu_test_base.h"
28#include "xla/service/hlo_module_util.h"
29#include "xla/tests/hlo_pjrt_interpreter_reference_mixin.h"
30#include "xla/tsl/platform/test.h"
31#include "tsl/platform/path.h"
32
33namespace xla {
34namespace {
35
36using ConvolutionHloTest =
37 HloPjRtInterpreterReferenceMixin<gpu::HloPjRtGpuTestBase>;
38
39TEST_F(ConvolutionHloTest, TestCudnnConvInt8x32) {
40 // This convolution should be transformed to "cudnn-conv" and vectorized as
41 // …
48:62… indow={size=3x3 pad=1_1x1_1}, dim_labels=b01f_o01i->b01f
49})";
50 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0, 0}));
51}
52
53TEST_F(ConvolutionHloTest, TestCudnnConvInt8x32Bias) {
54 // cudnnConvolutionBiasActiv …
175:18… terHloTest, TestCudnnReorderFilter) {
176 // Test that the bitcast-transpose-bitcast with reverse-engineered reordering
177 // works the same way as cud …