Skip to content

Commit

Permalink
[MPS][BE] Remove stale TODO (#118902)
Browse files Browse the repository at this point in the history
And use convenient methods

TODO was added by an accidental copy-n-paste of code from #82315 into  #88532

Pull Request resolved: #118902
Approved by: https://github.com/kit1980
  • Loading branch information
malfet authored and pytorchmergebot committed Feb 1, 2024
1 parent a16df1d commit 844a76e
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions aten/src/ATen/native/mps/operations/Unique.mm
Original file line number Diff line number Diff line change
Expand Up @@ -186,11 +186,7 @@
@autoreleasepool {
string key = getUniqueKey(self.scalar_type(), self.sizes(), return_inverse, return_counts, consecutive, dim);
return LookUpOrCreateCachedGraph<UniqueCachedGraph>(key, [&](auto mpsGraph, auto newCachedGraph) {
// Workaround for MPSShaderLibrary bug
// TODO: Remove once https://github.com/pytorch/pytorch/issues/82305 is resolved
auto inputType = getMPSScalarType(self.scalar_type());
newCachedGraph->inputTensor_ = mpsGraphRankedPlaceHolder(mpsGraph, inputType, getMPSShape(self.sizes()));

newCachedGraph->inputTensor_ = mpsGraphRankedPlaceHolder(mpsGraph, getMPSScalarType(self), getMPSShape(self));
auto outputTensors = buildUniqueGraph(self, newCachedGraph, return_inverse, return_counts, consecutive, dim);

newCachedGraph->outputTensor_ = outputTensors[0];
Expand Down

0 comments on commit 844a76e

Please sign in to comment.