Merge pull request #333 from devreal/dot-remove-keytype

Remove key type from Dot if types are disabled

Author: devreal

.github/workflows/gpu.yml

@@ -50,7 +50,7 @@ jobs:
           -DCMAKE_C_COMPILER=/home/evaleev/code/install/gcc/12/bin/gcc \
           -DCMAKE_CUDA_HOST_COMPILER=/home/evaleev/code/install/gcc/12/bin/g++ \
           -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc \
-          -DCMAKE_CUDA_ARCHITECTURES=50 \
+          -DCMAKE_CUDA_ARCHITECTURES=native \
           -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda \
           -DCMAKE_BUILD_TYPE=Release \
           -DBUILD_SHARED_LIBS=OFF \
@@ -90,7 +90,7 @@ jobs:
           -DCMAKE_C_COMPILER=/home/evaleev/code/install/gcc/12/bin/gcc \
           -DCMAKE_CUDA_HOST_COMPILER=/home/evaleev/code/install/gcc/12/bin/g++ \
           -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc \
-          -DCMAKE_CUDA_ARCHITECTURES=50 \
+          -DCMAKE_CUDA_ARCHITECTURES=native \
           -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda \
           -DCMAKE_PREFIX_PATH="${{github.workspace}}/install" \
           || (cat CMakeFiles/CMakeConfigureLog.yaml && exit 1)

ttg/ttg/util/dot.h

@@ -72,8 +72,7 @@ namespace ttg {
           if (in) {
             if (count != in->get_index()) throw "ttg::Dot: lost count of ins";
             if (disable_type) {
-              ttss << " <in" << count << ">"
-                   << escape(in->get_name());
+              ttss << " <in" << count << ">" << escape(in->get_name());
             } else {
               ttss << " <in" << count << ">"
                    << " " << escape("<" + in->get_key_type_str() + "," + in->get_value_type_str() + ">") << " "
@@ -97,8 +96,7 @@ namespace ttg {
           if (out) {
             if (count != out->get_index()) throw "ttg::Dot: lost count of outs";
             if (disable_type) {
-              ttss << " <out" << count << ">"
-                   << out->get_name();
+              ttss << " <out" << count << ">" << out->get_name();
             } else {
               ttss << " <out" << count << ">"
                    << " " << escape("<" + out->get_key_type_str() + "," + out->get_value_type_str() + ">") << " "