Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Conversation

@sxjscience
Copy link
Member

Description

Fix the cmake template to reflect the recent change of cutensor. @MoisesHer

Checklist

Essentials

  • PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL], [FEATURE], [DOC], etc)
  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage
  • Code is well-documented

@sxjscience sxjscience requested review from leezu and szha as code owners October 12, 2020 01:19
@mxnet-bot
Copy link

Hey @sxjscience , Thanks for submitting the PR
All tests are already queued to run once. If tests fail, you can trigger one or more tests again with the following commands:

  • To trigger all jobs: @mxnet-bot run ci [all]
  • To trigger specific jobs: @mxnet-bot run ci [job1, job2]

CI supported jobs: [centos-gpu, unix-cpu, windows-cpu, windows-gpu, clang, centos-cpu, sanity, edge, website, unix-gpu, miscellaneous]


Note:
Only following 3 categories can trigger CI :PR Author, MXNet Committer, Jenkins Admin.
All CI tests must pass before the PR can be merged.

@sxjscience
Copy link
Member Author

sxjscience commented Oct 13, 2020

To install cutensor, we need to use the following command:

sudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub	
sudo apt update	
sudo apt -y install libcutensor1 libcutensor-dev libcutensor-doc	

# In addition, we can choose to link libcutensor.so to the 10.2 version
sudo update-alternatives --set libcutensor.so.1.2.1 /usr/lib/x86_64-linux-gnu/libcutensor/10.2/libcutensor.so.1.2.1

@sxjscience sxjscience changed the title [Cmake] Create FindCUTENSOR.cmake [Fix][CMake] Create FindCUTENSOR.cmake Oct 13, 2020
@sxjscience sxjscience added the CMake CMake related bugs/issues/improvements label Oct 13, 2020
update cmake config

Update CMakeLists.txt

debug

Update FindCUTENSOR.cmake

Update linux_gpu.cmake

remove int types to support cuda 10

Update CMakeLists.txt

update

Update libinfo.cc

Revert "Update CMakeLists.txt"

This reverts commit 70afbf0.

Revert "Update linux_gpu.cmake"

This reverts commit bc5d2fc.

Revert "Revert "Update linux_gpu.cmake""

This reverts commit 96d5f43.

Revert "Revert "Update CMakeLists.txt""

This reverts commit ec20b33.
@sxjscience
Copy link
Member Author

I've checked manually and find that cutensor is used.

@leezu leezu merged commit ce1e682 into apache:master Oct 14, 2020
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

CMake CMake related bugs/issues/improvements

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants