@@ -253,7 +253,6 @@ TEST_F(TcCudaMapperTest, BatchTripleHadamard) {
253
253
at::Tensor V = at::CUDA (at::kFloat ).rand ({B, D});
254
254
at::Tensor W = at::CUDA (at::kFloat ).rand ({B, D});
255
255
std::vector<at::Tensor> inputs = {U, V, W};
256
- std::vector<at::Tensor> outputs;
257
256
258
257
static constexpr auto TC = R"TC(
259
258
def batch_triple_hadamard(float(B, D) U, float(B, D) V, float(B, D) W) -> (Z) {
@@ -283,7 +282,6 @@ TEST_F(TcCudaMapperTest, TensorDot) {
283
282
at::Tensor I0 = at::CUDA (at::kFloat ).rand ({N, C1, C2, H, W});
284
283
at::Tensor I1 = at::CUDA (at::kFloat ).rand ({N, C2, C3, H, W});
285
284
std::vector<at::Tensor> inputs = {I0, I1};
286
- std::vector<at::Tensor> outputs;
287
285
288
286
static constexpr auto TC = R"TC(
289
287
def tensordot(float(N, C1, C2, H, W) I0, float(N, C2, C3, H, W) I1) -> (O) {
@@ -309,7 +307,6 @@ TEST_F(TcCudaMapperTest, LUT) {
309
307
at::Tensor I =
310
308
at::CUDA (at::kFloat ).rand ({B, N}).mul_ (B).floor_ ().toType (at::kInt );
311
309
std::vector<at::Tensor> inputs = {LUT, I};
312
- std::vector<at::Tensor> outputs;
313
310
314
311
static constexpr auto TC = R"TC(
315
312
def fun(float(B, R) LUT, int32(B, N) I) -> (O) {
@@ -361,7 +358,6 @@ TEST_F(TcCudaMapperTest, DISABLED_SpatialBatchNormalization) {
361
358
at::Tensor rMeanIn = at::CUDA (at::kFloat ).rand ({C2});
362
359
at::Tensor rVarIn = at::CUDA (at::kFloat ).rand ({C2});
363
360
std::vector<at::Tensor> inputs = {momentum, eps, I, rMeanIn, rVarIn};
364
- std::vector<at::Tensor> outputs;
365
361
366
362
static constexpr auto TC = R"TC(
367
363
def spatial_batch_norm(
0 commit comments