Merge pull request #73 from lucasdelimanogueira/tmp

fix import distributed
2024-05-24 19:55:49 -03:00 · 2024-05-24 19:55:49 -03:00 · 5f92e6dc8c
commit 5f92e6dc8c
parent 2720f5c88c 29ee569d04
4 changed files with 1 additions and 1 deletions
--- a/build/distributed.o
+++ b/build/distributed.o
--- a/norch/csrc/distributed.cpp
+++ b/norch/csrc/distributed.cpp
@ -2,7 +2,6 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include "distributed.h"
-#include "tensor.h"
 #include "cuda.h"
 #include <mpi.h>
 #include <nccl.h>
--- a/norch/csrc/distributed.h
+++ b/norch/csrc/distributed.h
@ -1,5 +1,6 @@
 #ifndef DISTRIBUTED_H
 #define DISTRIBUTED_H
+#include "tensor.h"

 #define MPI_CHECK(cmd) do {                          \
  int e = cmd;                                      \
--- a/norch/libtensor.so
+++ b/norch/libtensor.so