Minor cstdint fix. (#204)

vbharadwaj-bk · web-flow · commit c3ffb671eaeb · 2026-06-14T10:23:20.000-07:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,8 @@
 ## Latest Changes
 
+### v0.6.8 (2026-06-14)
+Added `#include <cstdint>` to all C++ extension headers and sources. 
+
 ### v0.6.7 (2026-06-13)
 
 **Added**:
diff --git a/openequivariance/openequivariance/extension/backend/backend_cuda.hpp b/openequivariance/openequivariance/extension/backend/backend_cuda.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include <cstdint>
 #include <nvrtc.h>
 #include <cuda.h>
 #include <cuda_runtime.h>
diff --git a/openequivariance/openequivariance/extension/backend/backend_hip.hpp b/openequivariance/openequivariance/extension/backend/backend_hip.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include <cstdint>
 #include <hip/hip_runtime.h>
 #include <hip/hiprtc.h>
 
diff --git a/openequivariance/openequivariance/extension/group_mm.hpp b/openequivariance/openequivariance/extension/group_mm.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include <cstdint>
 #include <memory>
 #include <stdexcept>
 
diff --git a/openequivariance/openequivariance/extension/json11/json11.cpp b/openequivariance/openequivariance/extension/json11/json11.cpp
@@ -25,6 +25,7 @@
 #include <cstdlib>
 #include <cstdio>
 #include <limits>
+#include <cstdint>
 
 namespace json11 {
 
diff --git a/openequivariance/openequivariance/extension/json11/json11.hpp b/openequivariance/openequivariance/extension/json11/json11.hpp
@@ -55,6 +55,7 @@
 #include <map>
 #include <memory>
 #include <initializer_list>
+#include <cstdint>
 
 #ifdef _MSC_VER
     #if _MSC_VER <= 1800 // VS 2013
diff --git a/openequivariance/openequivariance/extension/libtorch_tp_jit.cpp b/openequivariance/openequivariance/extension/libtorch_tp_jit.cpp
@@ -1,3 +1,4 @@
+#include <cstdint>
 #include <pybind11/pybind11.h>
 
 #ifdef CUDA_BACKEND
diff --git a/openequivariance/openequivariance/extension/libtorch_tp_jit_stable.cpp b/openequivariance/openequivariance/extension/libtorch_tp_jit_stable.cpp
@@ -1,5 +1,6 @@
 #define USE_CUDA
 
+#include <cstdint>
 #include <torch/csrc/stable/accelerator.h>
 #include <torch/csrc/stable/library.h>
 #include <torch/csrc/stable/ops.h>
diff --git a/openequivariance/openequivariance/extension/stubs/stream.cpp b/openequivariance/openequivariance/extension/stubs/stream.cpp
@@ -1,3 +1,4 @@
+#include <cstdint>
 #include <torch/csrc/inductor/aoti_torch/c/shim.h>
 
 extern "C" {
diff --git a/openequivariance/openequivariance/extension/test/load_aoti.cpp b/openequivariance/openequivariance/extension/test/load_aoti.cpp
@@ -1,3 +1,4 @@
+#include <cstdint>
 #include <vector>
 #include <torch/torch.h>
 #include <torch/csrc/inductor/aoti_package/model_package_loader.h>
diff --git a/openequivariance/openequivariance/extension/torch_core.hpp b/openequivariance/openequivariance/extension/torch_core.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include <cstdint>
 #include <initializer_list>
 #include <iostream>
 #include <memory>
diff --git a/openequivariance/pyproject.toml b/openequivariance/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "scikit_build_core.build"
 
 [project]
 name = "openequivariance"
-version = "0.6.7"
+version = "0.6.8"
 authors = [
   { name="Austin Glover" },
   { name="Vivek Bharadwaj" },
diff --git a/openequivariance_extjax/pyproject.toml b/openequivariance_extjax/pyproject.toml
@@ -8,7 +8,7 @@ build-backend = "scikit_build_core.build"
 
 [project]
 name = "openequivariance_extjax"
-version = "0.6.7"
+version = "0.6.8"
 
 authors = [
   { name="Austin Glover" },

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+#include <cstdint>`
`1`	`2`	`#include <pybind11/pybind11.h>`
`2`	`3`
`3`	`4`	`#ifdef CUDA_BACKEND`
Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+#include <cstdint>`
`1`	`2`	`#include <torch/csrc/inductor/aoti_torch/c/shim.h>`
`2`	`3`
`3`	`4`	`extern "C" {`
Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+#include <cstdint>`
`1`	`2`	`#include <vector>`
`2`	`3`	`#include <torch/torch.h>`
`3`	`4`	`#include <torch/csrc/inductor/aoti_package/model_package_loader.h>`