Skip to content

Commit 9795769

Browse files
committed
add xccl funs
1 parent 66a33d4 commit 9795769

File tree

3 files changed

+318
-18
lines changed

3 files changed

+318
-18
lines changed

backends/metax_gpu/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ file(
5656
${PADDLE_SOURCE_DIR}/paddle/phi/backends/gpu/cuda/cuda_info.cc
5757
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/math_function.cu
5858
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/index_add_kernel.cu
59-
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/eye_kernel.cu
59+
#${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/eye_kernel.cu
6060
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/bincount_kernel.cu
6161
${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/uniform_kernel.cu)
6262

backends/metax_gpu/build.sh

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,16 @@
1414
# See the License for the specific language governing permissions and
1515
# limitations under the License.
1616

17-
set -e
17+
set -ex
1818

19+
CODE_ROOT="$( cd "$( dirname "${BASH_SOURCE[0]}")/" && pwd )"
20+
export CODE_ROOT
1921
echo "uninstall paddle-metax-gpu..."
2022
pip uninstall -y paddle-metax-gpu
2123

24+
cd ${CODE_ROOT}/../../Paddle
25+
git submodule update --init
26+
cd ${CODE_ROOT}
2227
if [ ! -d build ]; then
2328
echo "build directory not found, creating..."
2429
mkdir build

0 commit comments

Comments
 (0)