We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e5b1443 commit 26993edCopy full SHA for 26993ed
model.cpp
@@ -8,6 +8,10 @@
8
#include "gsplat.hpp"
9
#include "utils.hpp"
10
11
+#ifdef USE_MPS
12
+#include <torch/mps.h>
13
+#endif
14
+
15
#ifdef USE_HIP
16
#include <c10/hip/HIPCachingAllocator.h>
17
#elif defined(USE_CUDA)
@@ -178,6 +182,9 @@ torch::Tensor Model::forward(Camera& cam, int step){
178
182
rgbs = SphericalHarmonicsCPU::apply(degreesToUse, viewDirs, colors);
179
183
}else{
180
184
#if defined(USE_HIP) || defined(USE_CUDA) || defined(USE_MPS)
185
+ #ifdef USE_MPS
186
+ torch::mps::synchronize();
187
+ #endif
181
188
rgbs = SphericalHarmonics::apply(degreesToUse, viewDirs, colors);
189
#endif
190
}
0 commit comments