Skip to content
This repository was archived by the owner on Jul 1, 2023. It is now read-only.

Commit c54929a

Browse files
authored
Remove use of xla::ValidateEinsumNumericDimensions (#1066)
tensorflow/tensorflow@116792d removes the function `xla::ValidateEinsumNumericDimensions` because all expressible op configurations are valid, and thus this function effectively always returns `true`. This change updates the S4TF usage to remove the function call, and always return true.
1 parent c1d8225 commit c54929a

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

Sources/x10/xla_tensor/ops/einsum.cpp

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -75,12 +75,7 @@ bool Einsum::SupportsEquation(const std::string& equation, xla::int64 x_rank,
7575
if (!einsum_config_numeric_or_err.ok()) {
7676
return false;
7777
}
78-
auto einsum_config_numeric = einsum_config_numeric_or_err.ConsumeValueOrDie();
79-
auto validation_status = xla::ValidateEinsumNumericDimensions(
80-
/*x_config=*/einsum_config_numeric[0],
81-
/*y_config=*/einsum_config_numeric[1],
82-
/*output_config=*/einsum_config_numeric[2]);
83-
return validation_status.ok();
78+
return true;
8479
}
8580

8681
} // namespace ops

0 commit comments

Comments
 (0)