fixed review comments

Signed-off-by: majing <[email protected]>
intel · Jul 22, 2024 · 0063a00 · 0063a00
1 parent c7f7ccb
commit 0063a00
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/ATen/native/xpu/Histogram.cpp b/src/ATen/native/xpu/Histogram.cpp
@@ -122,7 +122,6 @@ static Tensor& histogramdd_out(
     bool density,
     Tensor& hist,
     Tensor& bin_edges) {
-  globalContext().alertNotDeterministic("histogram_bin_xpu");
   histogramdd_check_inputs(self, bins, weight);
   histogramdd_prepare_out(self, bins, hist, bin_edges);
 
@@ -168,7 +167,6 @@ std::tuple<Tensor&, Tensor&> XPUNativeFunctions::histogram_out(
     bool density,
     Tensor& hist,
     Tensor& bin_edges) {
-  globalContext().alertNotDeterministic("histogram_bin_count_xpu");
   Tensor reshaped_self = self.reshape({self.numel()});
   std::optional<Tensor> reshaped_weight = weight.has_value()
       ? weight.value().reshape({weight.value().numel()})

diff --git a/src/ATen/native/xpu/sycl/HistogramddKernels.cpp b/src/ATen/native/xpu/sycl/HistogramddKernels.cpp
@@ -151,6 +151,7 @@ void histogramdd_kernel(
     bool density,
     Tensor& hist,
     const Tensor& bin_edges_) {
+  globalContext().alertNotDeterministic("histogramdd_kernel_xpu");
   hist.fill_(0);
   Tensor bin_edges = bin_edges_.contiguous();
   AT_DISPATCH_FLOATING_TYPES_AND2(
@@ -180,6 +181,7 @@ void histogramdd_linear_kernel(
     bool density,
     Tensor& hist,
     Tensor& out_bin_edges) {
+  globalContext().alertNotDeterministic("histogramdd_linear_kernel_xpu");
   hist.fill_(0);
 
   // default range for empty input